Spaces:

TZTestAnalysis
/

OrchAId

Sleeping

jatamura commited on Jun 4, 2025

Commit

f368cd3

verified ·

1 Parent(s): f524400

Update python_utils/get_model.py

Files changed (1) hide show

python_utils/get_model.py CHANGED Viewed

@@ -25,7 +25,7 @@ def load_model():
     ## define relevant parameters
     cfg = get_cfg()
     cfg.merge_from_file("./configs/test_model_config.yaml")
-    cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("./configs/test_model_config.yaml")
     if not torch.cuda.is_available():
         cfg.MODEL.DEVICE = "cpu"
     else:

     ## define relevant parameters
     cfg = get_cfg()
     cfg.merge_from_file("./configs/test_model_config.yaml")
+    cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("https://huggingface.co/TZProject/test_model/model_final.pth")
     if not torch.cuda.is_available():
         cfg.MODEL.DEVICE = "cpu"
     else: