Spaces:

innat
/

VideoMAE

Sleeping

innat commited on Oct 7, 2023

Commit

fbc7dad

•

1 Parent(s): 295f23f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,8 +14,8 @@ from labels import K400_label_map, SSv2_label_map, UCF_label_map
 MODEL_CACHE = {}
 MODELS = {
     'K400': [
-        '/mnt/c/Users/innat/Desktop/videomae/K400/small/TFVideoMAE_S_K400_16x224_FT',
-        '/mnt/c/Users/innat/Desktop/videomae/K400/small/TFVideoMAE_S_K400_16x224_PT'
         ],
     'SSv2': [],
     'UCF' : []
@@ -121,8 +121,8 @@ gr.Interface(
     ],
     examples=[
         ["examples/k400.mp4"],
-        ["examples/ssv2.mp4"],
-        ["examples/ucf.avi"],
     ],
     title="VideoMAE",
 ).launch()

 MODEL_CACHE = {}
 MODELS = {
     'K400': [
+        'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
+        'innat/videomae/TFVideoMAE_S_K400_16x224_PT'
         ],
     'SSv2': [],
     'UCF' : []
     ],
     examples=[
         ["examples/k400.mp4"],
+        ["examples/k400.mp4"],
+        ["examples/k400.mp4"],
     ],
     title="VideoMAE",
 ).launch()