Fixing model location

ml4ai · Apr 22, 2024 · a4bf002 · a4bf002
1 parent be20725
commit a4bf002
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/retrieve_model_ci.py b/scripts/retrieve_model_ci.py
@@ -14,7 +14,7 @@ def retrieve_model(model_path=None) -> str:
         str: Path to the loaded model file.
     """
     cwd = Path(__file__).parents[0]
-    REPO_NAME = "vraymond/img2mml"
+    REPO_NAME = "vraymond6/img2mml"
     MODEL_NAME = "cnn_xfmer_arxiv_im2mml_with_fonts_boldface_best.pt"
     # If the model path is none or doesn't exist, the default model will be downloaded from server.
     if model_path is None or not os.path.exists(model_path):