Update app.py

app.py

@@ -123,19 +123,16 @@ def main():
     datasets = ['K400', 'SSv2', 'UCF']
 
     ALL_MODELS = [
-        'TFVideoMAE_S_K400_16x224',
-        'TFVideoMAE_B_K400_16x224',
         'TFVideoMAE_L_K400_16x224',
-        'TFVideoMAE_S_SSv2_16x224',
         'TFVideoMAE_B_SSv2_16x224',
         'TFVideoMAE_B_UCF_16x224',
     ]
 
     
     sample_example = [
-        ["examples/k400.mp4",  1, 0.9],
-        ["examples/ucf.mp4",  1, 0.8],
-        ["examples/k400.mp4",  1, 0.7]
+        ["examples/k400.mp4",  ALL_MODELS[0], 0.9],
+        ["examples/k400.mp4",  ALL_MODELS[2], 0.8]
+        ["examples/ucf.mp4",  ALL_MODELS[1], 0.7],
     ]
 
     iface = gr.Interface(