Fix num_frames parameter used by both model and train params

app.py

@@ -260,16 +260,15 @@ def predict_on_images(data_files: list, mask_ratio: float, yaml_file_path: str,
     with open(yaml_file_path, 'r') as f:
         params = yaml.safe_load(f)
 
+    model_params = params["model_args"]
     # data related
     train_params = params["train_params"]
-    num_frames = train_params['num_frames']
+    num_frames = model_params['num_frames']
     img_size = train_params['img_size']
     bands = train_params['bands']
     mean = train_params['data_mean']
     std = train_params['data_std']
 
-    model_params = params["model_args"]
-
     batch_size = 8
 
     mask_ratio = train_params['mask_ratio'] if mask_ratio is None else mask_ratio