Spaces:
Runtime error
Runtime error
thinh-huynh-re
commited on
Commit
·
53628ab
1
Parent(s):
c8b45a2
Fix
Browse files
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 |
)
|