Update app.py
Browse files
app.py
CHANGED
@@ -55,10 +55,10 @@ def get_model(model_type):
|
|
55 |
print('innat/videomae/' + model_type + '_FT')
|
56 |
|
57 |
ft_path = hf_hub_download(
|
58 |
-
repo_id=
|
59 |
)
|
60 |
pt_path = hf_hub_download(
|
61 |
-
repo_id=
|
62 |
)
|
63 |
|
64 |
ft_model = keras.models.load_model(ft_path)
|
|
|
55 |
print('innat/videomae/' + model_type + '_FT')
|
56 |
|
57 |
ft_path = hf_hub_download(
|
58 |
+
repo_id='innat/videomae', filename=model_type + '_FT', repo_type="model"
|
59 |
)
|
60 |
pt_path = hf_hub_download(
|
61 |
+
repo_id='innat/videomae', filename=model_type + '_PT', repo_type="model"
|
62 |
)
|
63 |
|
64 |
ft_model = keras.models.load_model(ft_path)
|