2025-07-31 21:26 🐛

app.py

@@ -46,7 +46,8 @@ def load_model(variant: str, dataset: str = "ShanghaiTech B", metric: str = "mae
 
     model = get_model(model_info_path=model_info_path)
     model.eval()
-    loaded_model = model
+    # loaded_model = model
+    return model
 
 
 def _calc_size(
@@ -254,7 +255,7 @@ def predict(image: Image.Image, variant_dataset: str, metric: str):
         dataset_name = "nwpu"
 
     if loaded_model is None:
-        load_model(variant=variant, dataset=dataset_name, metric=metric)
+        loaded_model = load_model(variant=variant, dataset=dataset_name, metric=metric)
 
     if not hasattr(loaded_model, "input_size"):
         if dataset_name == "sha":