Update app.py

app.py

@@ -16,8 +16,14 @@ MODELS = {
         'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
         'innat/videomae/TFVideoMAE_S_K400_16x224_PT'
         ],
-    'SSv2': [],
-    'UCF' : []
+    'SSv2': [
+        'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
+        'innat/videomae/TFVideoMAE_S_K400_16x224_PT'
+        ],
+    'UCF' : [
+        'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
+        'innat/videomae/TFVideoMAE_S_K400_16x224_PT'
+        ]
 }
 
 
@@ -45,6 +51,9 @@ def video_to_gif(video_array, gif_filename):
 
 
 def get_model(data_type):
+    print()
+    print('-------------------- ', data_type)
+    print()
     ft_model = keras.models.load_model(MODELS[data_type][0])
     pt_model = keras.models.load_model(MODELS[data_type][1])
     label_map = {v: k for k, v in K400_label_map.items()}