Fix img_size

app.py

@@ -264,7 +264,7 @@ def predict_on_images(data_files: list, mask_ratio: float, yaml_file_path: str,
     # data related
     train_params = params["train_params"]
     num_frames = model_params['num_frames']
-    img_size = train_params['img_size']
+    img_size = model_params['img_size']
     bands = train_params['bands']
     mean = train_params['data_mean']
     std = train_params['data_std']