Update script.py

script.py

@@ -35,6 +35,9 @@ def preprocess(file_like):
             if current_memory >= MAX_MEMORY:
                 break
 
+    video_tensor = torch.stack(frames)
+    return video_tensor
+
 
 class Model(torch.nn.Module):
     def __init__(self):
@@ -90,7 +93,7 @@ for el in tqdm.tqdm(dataset_remote):
     except Exception as e:
         print(e)
         print("failed", el["id"])
-        raise e
+        # raise e
         out.append(dict(id=el["id"]))
 
 # save the final result and that's it