Updated demo

app.py

@@ -80,8 +80,10 @@ def predict_on_images(data_file: str | Path, config_path: str, checkpoint: str):
 
     # Load model ---------------------------------------------------------------------------------
 
+    print(f'Loading model')
     lightning_model = LightningInferenceModel.from_config(config_path, checkpoint)
-    img_size = 256  # Size of Sen1Floods11
+    img_size = 512  # Size of BurnScars
+    print(f'Model loaded')
 
     # Loading data ---------------------------------------------------------------------------------