Update app.py

app.py

@@ -36,7 +36,7 @@ def get_model(model_type):
     
     model = keras.models.load_model(model_type)
 
-    if 'K400' in model_type:
+    if 'W877' in model_type:
         data_type = 'K400'
     else:
         data_type = 'SSv2'
@@ -76,9 +76,7 @@ def main():
                 label="Model"
             )
         ],
-        outputs=[
-            gr.Label(num_top_classes=3, label='scores'),
-        ],
+        outputs=gr.Label(num_top_classes=3, label='scores'),
         examples=sample_example,
         title="VideoSwin: Video Swin Transformer",
         description="Keras reimplementation of <a href='https://github.com/innat/VideoSwin'>VideoSwin</a> is presented here."