diff --git a/.gitignore b/.gitignore index 2722a76..4fb7c29 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,3 @@ -DocRes -DocScanner -venv +* +!.gitignore +!api.py diff --git a/api.py b/api.py index bd89034..bbdd276 100644 --- a/api.py +++ b/api.py @@ -62,7 +62,7 @@ if _docres_model_env: if not DOCRES_MODEL_PATH.is_absolute(): DOCRES_MODEL_PATH = (SCAN_DIR / DOCRES_MODEL_PATH).resolve() else: - DOCRES_MODEL_PATH = DOCREST_DIR / "checkpoints" / "docres.pkl" + DOCRES_MODEL_PATH = DOCREST_DIR / "data/weights" / "docres.pkl" _lock = threading.Lock()