Fixed normalization error when only 1 image is sent to endpoint

handler.py

@@ -227,8 +227,8 @@ class EndpointHandler:
                     # NORMALIZE
                     if batch_emb.ndim > 1:
                         batch_emb = batch_emb.T / np.linalg.norm(batch_emb, axis=1)
-                    # transpose back to (21, 512)
-                    batch_emb = batch_emb.T.tolist()
+                        # transpose back to (21, 512)
+                        batch_emb = batch_emb.T.tolist()
                 embedding_end_time = timeit.default_timer()
                 self.logger.info(
                     f"Embedding calculation took {embedding_end_time - embedding_start_time} seconds"