Removed troch.stack()

handler.py

@@ -111,12 +111,12 @@ class EndpointHandler:
         frame_embedding = self.model.get_video_features(**frame_preprocessed)
 
         # Stack the list of frame embeddings into a single tensor
-        self.logger.info("Stacking embeddings into a single tensor.")
-        tensor = torch.stack(frame_embedding)
+        # self.logger.info("Stacking embeddings into a single tensor.")
+        # tensor = torch.stack(frame_embedding)
 
         # detach text emb from graph, move to CPU, and convert to numpy array
         self.logger.info("Squeezing tensor")
-        batch_emb = tensor.squeeze(0)
+        batch_emb = frame_embedding.squeeze(0)
 
         self.logger.info("Converting into numpy array")
         batch_emb = batch_emb.cpu().detach().numpy()