Spaces:

Andy1621
/

UniFormerV2_mit_demo

Runtime error

SakuraD commited on Jan 6, 2023

Commit

0439c37

1 Parent(s): ad30d2b

init

Files changed (3) hide show

.DS_Store ADDED Viewed

Binary file (6.15 kB). View file

app.py CHANGED Viewed

@@ -91,7 +91,7 @@ def inference(video):
     prediction = model(inputs)
     prediction = F.softmax(prediction, dim=1).flatten()
-    return {mitv1_id_to_classname[str(i)]: float(prediction[i]) for i in range(400)}
 def set_example_video(example: list) -> dict:

     prediction = model(inputs)
     prediction = F.softmax(prediction, dim=1).flatten()
+    return {mitv1_id_to_classname[str(i)]: float(prediction[i]) for i in range(339)}
 def set_example_video(example: list) -> dict:

clapping.mp4 CHANGED Viewed

Binary files a/clapping.mp4 and b/clapping.mp4 differ