Update app.py

app.py

@@ -47,12 +47,19 @@ def tube_mask_generator(mask_ratio):
 
 
 def get_model(model_type):
-    # 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 = keras.utils.get_file(
+        origin=f'https://github.com/innat/VideoMAE/releases/download/v1.1/{model_type}_FT.zip',
+    )
+    pt_path = keras.utils.get_file(
+        origin=f'https://github.com/innat/VideoMAE/releases/download/v1.1/{model_type}_PT.zip',
+    )
+
+    with zipfile.ZipFile(ft_path, 'r') as zip_ref:
+        zip_ref.extractall('./')
+    
+    with zipfile.ZipFile(pt_path, 'r') as zip_ref:
+        zip_ref.extractall('./')
+
     ft_model = keras.models.load_model(model_type + '_FT')
     pt_model = keras.models.load_model(model_type + '_PT')