Fixed normalization error when only 1 image is sent to endpoint

handler.py

@@ -225,8 +225,8 @@ class EndpointHandler:
 
                     batch_emb = batch_emb.cpu().detach().numpy()
                     # NORMALIZE
-
-                    batch_emb = batch_emb.T / np.linalg.norm(batch_emb, axis=1)
+                    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()
                 embedding_end_time = timeit.default_timer()