Added normalization using torch and shape check

handler.py

@@ -118,9 +118,15 @@ 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)
+        # Check the shape of the tensor
+        self.logger.info(f"Shape of the batch_emb tensor: {batch_emb.shape}")
+
+        # Normalize the embeddings if it's a 2D tensor
+        if batch_emb.dim() == 2:
+            self.logger.info("Normalizing embeddings")
+            batch_emb = torch.nn.functional.normalize(batch_emb, p=2, dim=1)
+        else:
+            self.logger.info("Skipping normalization due to tensor shape")
 
         self.logger.info("Converting into numpy array")
         batch_emb = batch_emb.cpu().detach().numpy()