Spaces:

innat
/

VideoMAE

Running

innat commited on Oct 8, 2023

Commit

0b2cebb

1 Parent(s): 2a166f0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,16 +13,16 @@ from labels import K400_label_map, SSv2_label_map, UCF_label_map
 MODELS = {
     'K400': [
-        'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
-        'innat/videomae/TFVideoMAE_S_K400_16x224_PT'
         ],
     'SSv2': [
-        'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
-        'innat/videomae/TFVideoMAE_S_K400_16x224_PT'
         ],
     'UCF' : [
         'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
-        'innat/videomae/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' : [
         'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
+        './TFVideoMAE_S_K400_16x224_PT'
         ]
 }