2025-07-31 21:17 🐛

Fixed a bug in app.py

app.py

@@ -257,6 +257,16 @@ def predict(image: Image.Image, variant_dataset: str, metric: str):
         
         load_model(variant=variant, dataset=dataset_name, metric=metric)
 
+        if not hasattr(loaded_model, "input_size"):
+            if dataset_name == "sha":
+                loaded_model.input_size = 224
+            elif dataset_name == "shb":
+                loaded_model.input_size = 448
+            elif dataset_name == "qnrf":
+                loaded_model.input_size = 672
+            elif dataset_name == "nwpu":
+                loaded_model.input_size = 672
+
     loaded_model.to(device)
 
     # Preprocess the image