Update app.py

app.py

@@ -12,20 +12,20 @@ from utils import IMAGENET_MEAN, IMAGENET_STD, num_frames, patch_size, input_siz
 from labels import K400_label_map, SSv2_label_map, UCF_label_map
 
 
-MODELS = {
-    'K400': [
-        './TFVideoMAE_S_K400_16x224_FT',
-        './TFVideoMAE_S_K400_16x224_PT'
-        ],
-    'SSv2': [
-        './TFVideoMAE_S_K400_16x224_FT',
-        './TFVideoMAE_S_K400_16x224_PT'
-        ],
-    'UCF' : [
-        './TFVideoMAE_S_K400_16x224_FT',
-        './TFVideoMAE_S_K400_16x224_PT'
-        ]
-}
+# MODELS = {
+#     'K400': [
+#         './TFVideoMAE_S_K400_16x224_FT',
+#         './TFVideoMAE_S_K400_16x224_PT'
+#         ],
+#     'SSv2': [
+#         './TFVideoMAE_S_K400_16x224_FT',
+#         './TFVideoMAE_S_K400_16x224_PT'
+#         ],
+#     'UCF' : [
+#         './TFVideoMAE_S_K400_16x224_FT',
+#         './TFVideoMAE_S_K400_16x224_PT'
+#         ]
+# }
 
 LABEL_MAPS = {
     'K400': K400_label_map,
@@ -51,18 +51,15 @@ def tube_mask_generator(mask_ratio):
 
 
 def get_model(model_type):
-    print('---------------------')
-    print('innat/videomae/' + model_type + '_FT')
+    # ft_path = hf_hub_download(
+    #     repo_id='innat/videomae', filename=model_type + '_FT', repo_type="model"
+    # )
+    # pt_path = hf_hub_download(
+    #     repo_id='innat/videomae', filename=model_type + '_PT', repo_type="model"
+    # )
 
-    ft_path = hf_hub_download(
-        repo_id='innat/videomae', filename=model_type + '_FT', repo_type="model"
-    )
-    pt_path = hf_hub_download(
-        repo_id='innat/videomae', filename=model_type + '_PT', repo_type="model"
-    )
-
-    ft_model = keras.models.load_model(ft_path)
-    pt_model = keras.models.load_model(pt_path)
+    ft_model = keras.models.load_model(model_type + '_FT')
+    pt_model = keras.models.load_model(model_type + '_PT')
 
     if 'K400' in model_type:
         data_type = 'K400'
@@ -126,7 +123,6 @@ def main():
         'TFVideoMAE_B_UCF_16x224',
     ]
 
-    
     sample_example = [
         ["examples/k400.mp4",  ALL_MODELS[0], 0.9],
         ["examples/k400.mp4",  ALL_MODELS[1], 0.8],