Update app.py

app.py

@@ -84,7 +84,7 @@ def inference(video):
     prediction = model(inputs)
     prediction = F.softmax(prediction, dim=1).flatten()
 
-    return {kinetics_id_to_classname[str(i)]: float(prediction[i]) for i in range(1000)}
+    return {kinetics_id_to_classname[str(i)]: float(prediction[i]) for i in range(400)}
     
 
 inputs = gr.inputs.Video()