Update app.py

app.py

@@ -65,8 +65,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()}
     
-    ft_model.trainable = False
-    pt_model.trainable = False
     MODEL_CACHE[data_type] = (ft_model, pt_model, label_map)
 
     return ft_model, pt_model, label_map
@@ -77,6 +75,8 @@ def inference(video_file, dataset_type):
     frames = frame_sampling(container, num_frames=num_frames)
     bool_masked_pos_tf = tube_mask_generator()
     ft_model, pt_model, label_map = get_model(dataset_type)
+    ft_model.trainable = False
+    pt_model.trainable = False
 
     # inference on fine-tune model
     outputs_ft = ft_model(frames[None, ...], training=False)
@@ -114,7 +114,6 @@ gr.Interface(
             label='Dataset'
         ),
     ],
-
     outputs=[
         gr.Label(num_top_classes=3, label='confidence scores'),
         gr.Image(type="filepath", label='reconstructed masked autoencoder')