Spaces:

Luigi
/

Video-Human-Fall-Detection-with-TimeSformer

Sleeping

Luigi commited on Apr 14

Commit

413c57e

1 Parent(s): 4c9f334

bugfix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import gradio as gr
 from transformers import AutoFeatureExtractor, AutoModelForVideoClassification
 # Specify the model checkpoint for TimeSformer.
-MODEL_NAME = "microsoft/timesformer-base-finetuned-k400"
 def extract_frames(video_path, num_frames=16, target_size=(224, 224)):
     """

 from transformers import AutoFeatureExtractor, AutoModelForVideoClassification
 # Specify the model checkpoint for TimeSformer.
+MODEL_NAME = "facebook/timesformer-base-finetuned-k400"
 def extract_frames(video_path, num_frames=16, target_size=(224, 224)):
     """