2025-07-31 19:37 🐛

Fixed a bug in app.py

app.py

@@ -21,10 +21,10 @@ std = (0.229, 0.224, 0.225)
 alpha = 0.8
 EPS = 1e-8
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
+loaded_model = None
 
 pretrained_datasets = {
-    "ZIP-B": ["ShanghaiTech A", "ShanghaiTech B", "UCF-QNRF" "NWPU-Crowd"],
+    "ZIP-B": ["ShanghaiTech A", "ShanghaiTech B", "UCF-QNRF", "NWPU-Crowd"],
     "ZIP-S": ["ShanghaiTech A", "ShanghaiTech B", "UCF-QNRF"],
     "ZIP-T": ["ShanghaiTech A", "ShanghaiTech B", "UCF-QNRF"],
     "ZIP-N": ["ShanghaiTech A", "ShanghaiTech B", "UCF-QNRF"],