Update app.py

app.py

@@ -56,6 +56,7 @@ def inference(video_file, dataset_type, mask_ratio):
         frames[None, ...], bool_masked_pos_tf, outputs_pt
     )
 
+    # post process
     input_frame = denormalize(frames)
     input_mask = denormalize(mask[0] * frames)
     output_frame = denormalize(reconstruct_output)
@@ -81,11 +82,10 @@ def main():
             './TFVideoMAE_S_K400_16x224_PT'
             ],
         'UCF' : [
-            'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
+            './TFVideoMAE_S_K400_16x224_FT',
             './TFVideoMAE_S_K400_16x224_PT'
             ]
     }
-
     BENCHMARK_DATASETS = ['K400', 'SSv2', 'UCF']
     SAMPLE_EXAMPLES = [
         ["examples/k400.mp4", 'Kintetics-400'],
@@ -103,11 +103,11 @@ def main():
                 default=BENCHMARK_DATASETS[0],
                 label='Dataset', 
             ),
-            gr.inputs.Slider(
-                0.5,
-                1.0,
-                step=0.1,
-                default=0.7,
+            gr.Slider(
+                0,
+                1,
+                step=0.05,
+                default=0.5,
                 label='Mask Ratio'
             )
         ],