Update utils/classifier.py

utils/classifier.py

@@ -51,7 +51,7 @@ def load_wbc_classifier(weights_path: str | Path):
     )
 
     # Load checkpoint
-    ckpt = torch.load(weights_path, map_location=DEVICE)
+    ckpt = torch.load(weights_path, map_location=DEVICE,weights_only=False)
     if "model_state_dict" in ckpt:
         model.load_state_dict(ckpt["model_state_dict"])
     else: