Update app.py

app.py

@@ -39,7 +39,12 @@ def sample_frame_indices(
 # @st.cache_resource
 @st.experimental_singleton
 def load_model(model_name: str):
-    feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
+    if "k400" in model_name or "k600" in model_name:
+        feature_extractor = AutoFeatureExtractor.from_pretrained(
+            "MCG-NJU/videomae-base-finetuned-kinetics"
+        )
+    else:
+        feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
     model = TimesformerForVideoClassification.from_pretrained(model_name)
     return feature_extractor, model