Added normalization using torch

handler.py

@@ -118,14 +118,13 @@ class EndpointHandler:
         self.logger.info("Squeezing tensor")
         batch_emb = frame_embedding.squeeze(0)
 
+        # Normalize the embeddings
+        self.logger.info("Normalizing embeddings")
+        batch_emb = torch.nn.functional.normalize(batch_emb, p=2, dim=1)
+
         self.logger.info("Converting into numpy array")
         batch_emb = batch_emb.cpu().detach().numpy()
 
-        # NORMALIZE
-        # self.logger.info("Normalizing numpy array")
-        # batch_emb = batch_emb.T / np.linalg.norm(batch_emb, axis=1)
-        # transpose back to (21, 512)
-
         self.logger.info("Converting to list")
         batch_emb = batch_emb.tolist()