Update app.py

app.py

@@ -121,6 +121,17 @@ def inference(video_file, model_type, mask_ratio):
 
 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",  datasets[0], 0.9],
         ["examples/ucf.mp4",  datasets[1], 0.8],