requirements

app.py

@@ -219,7 +219,10 @@ def compute_video_embedding(video_reader, min_len):
     video_frames = torch.cat(video_frames, dim=0)
     video_frames = rearrange(video_frames, "f h w c -> f c h w")
     torch.cuda.empty_cache()
-    return encoded, video_frames
+    # Assuming video_embedding is a PyTorch tensor
+    video_embedding_safe = encoded.detach().cpu().numpy().tolist()
+    video_frames_safe = video_frames.detach().cpu().numpy().tolist()
+    return video_embedding_safe, video_frames_safe
 
 
 @spaces.GPU(duration=120)