Update app.py

app.py

@@ -32,7 +32,7 @@ LABEL_MAPS = {
 }
 
 def tube_mask_generator(mask_ratio):
-    mask_ratio=0.8
+    # mask_ratio=0.8
     window_size = (
         num_frames // 2, 
         input_size // patch_size[0], 
@@ -50,7 +50,7 @@ def tube_mask_generator(mask_ratio):
 
 
 def get_model(data_type):
-    data_type = 'K400'
+    # data_type = 'K400'
     ft_model = keras.models.load_model(MODELS[data_type][0])
     pt_model = keras.models.load_model(MODELS[data_type][1])
 
@@ -108,9 +108,9 @@ def inference(video_file, data_type, mask_ratio):
 def main():
     datasets = ['K400', 'SSv2', 'UCF']
     sample_example = [
-        ["examples/k400.mp4", 'Kintetics-400'],
-        ["examples/k400.mp4", 'Something-Something-V2'],
-        ["examples/k400.mp4", 'UCF101']
+        ["examples/k400.mp4",  datasets[0], 0.9],
+        ["examples/k400.mp4",  datasets[1], 0.8],
+        ["examples/k400.mp4",  datasets[2], 0.7]
     ]
 
     iface = gr.Interface(