Update app.py

app.py

@@ -11,17 +11,6 @@ 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
 
 
-def available_models():
-    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',
-    ]
-    return ALL_MODELS
-
 MODELS = {
     'K400': [
         './TFVideoMAE_S_K400_16x224_FT',
@@ -131,8 +120,8 @@ def main():
     
     sample_example = [
         ["examples/k400.mp4",  ALL_MODELS[0], 0.9],
-        ["examples/k400.mp4",  ALL_MODELS[2], 0.8]
-        ["examples/ucf.mp4",  ALL_MODELS[1], 0.7],
+        ["examples/k400.mp4",  ALL_MODELS[1], 0.8],
+        ["examples/ucf.mp4",  ALL_MODELS[2], 0.7],
     ]
 
     iface = gr.Interface(