Update app.py

app.py

@@ -11,7 +11,6 @@ from utils import IMAGENET_MEAN, IMAGENET_STD, num_frames, patch_size, input_siz
 from labels import K400_label_map, SSv2_label_map, UCF_label_map
 
 
-MODEL_CACHE = {}
 MODELS = {
     'K400': [
         'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
@@ -47,9 +46,6 @@ def video_to_gif(video_array, gif_filename):
 
 def get_model(data_type):
 
-    if data_type in MODEL_CACHE:
-        return MODEL_CACHE[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])
@@ -65,7 +61,6 @@ def get_model(data_type):
         pt_model = keras.models.load_model(MODELS[data_type][1])
         label_map = {v: k for k, v in UCF_label_map.items()}
     
-    MODEL_CACHE[data_type] = (ft_model, pt_model, label_map)
 
     return ft_model, pt_model, label_map
 
@@ -111,7 +106,7 @@ gr.Interface(
         gr.Video(type="file"),
         gr.Radio(
             ['K400', 'SSv2', 'UCF'], 
-            label='Dataset'
+            label='Dataset', value='K400'
         ),
     ],
     outputs=[