thinh-huynh-re commited on
Commit
53628ab
·
1 Parent(s): c8b45a2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def sample_frame_indices(
39
  # @st.cache_resource
40
  @st.experimental_singleton
41
  def load_model(model_name: str):
42
- if "k400" in model_name or "k600" in model_name:
43
  feature_extractor = AutoFeatureExtractor.from_pretrained(
44
  "MCG-NJU/videomae-base-finetuned-kinetics"
45
  )
 
39
  # @st.cache_resource
40
  @st.experimental_singleton
41
  def load_model(model_name: str):
42
+ if "base-finetuned-k400" in model_name or "base-finetuned-k600" in model_name:
43
  feature_extractor = AutoFeatureExtractor.from_pretrained(
44
  "MCG-NJU/videomae-base-finetuned-kinetics"
45
  )