Update app.py

app.py

@@ -45,23 +45,9 @@ def video_to_gif(video_array, gif_filename):
 
 
 def get_model(data_type):
-
-    if data_type=='K400':
-        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()}
-
-    elif data_type=='SSv2':
-        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 SSv2_label_map.items()}
-
-    elif data_type=='UCF':
-        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 UCF_label_map.items()}
-    
-
+    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()}
     return ft_model, pt_model, label_map