Updated demo

app.py

@@ -80,10 +80,8 @@ 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 = 512  # Size of BurnScars
-    print(f'Model loaded')
 
     # Loading data ---------------------------------------------------------------------------------