2025-07-31 20:41🐛

Fixed a bug in app.py

app.py

@@ -255,8 +255,7 @@ def predict(image: Image.Image, variant_dataset: str, metric: str):
         elif dataset == "NWPU-Crowd":
             dataset_name = "nwpu"
         
-        weight_path = f"Yiming-M/{variant}/checkpoints/{dataset_name}/best_{metric}.pth"
-        load_model(weight_path)
+        load_model(variant=variant, dataset=dataset_name, metric=metric)
 
     loaded_model.to(device)