mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-13 14:31:00 +08:00
feat: add default download model configuration
This commit is contained in:
1 parent
e8938d8c2a
commit
a4c7382bba
2 files changed
+9
-2
No files matched your search
@@ -412,7 +412,7 @@ def parse_args(arg_list: Optional[List[str]] = None):
|
||||
parser_check.set_defaults(func=check_install)
|
||||
|
||||
parser_download = subparser.add_parser("download_models", help="Download models")
|
||||
parser_download.add_argument("--config", type=str)
|
||||
parser_download.add_argument("--config", type=str, default=None)
|
||||
parser_download.set_defaults(func=download_models)
|
||||
|
||||
args = parser.parse_args(arg_list)
|
||||
|
||||
@@ -14,7 +14,14 @@ from .utils import mkdir
|
||||
root_dir = Path(__file__).resolve().parent.parent
|
||||
|
||||
|
||||
def download_models(config_path: Union[str, Path]) -> None:
|
||||
def download_models(config_path: Union[str, Path, None] = None) -> None:
|
||||
if config_path is None:
|
||||
config_path = root_dir / "config.yaml"
|
||||
else:
|
||||
config_path = Path(config_path).expanduser().resolve()
|
||||
if not config_path.is_file():
|
||||
raise FileNotFoundError(f"Config file not found: {config_path}")
|
||||
|
||||
cfg = ParseParams.load(config_path)
|
||||
|
||||
model_root_dir = cfg.Global.get("model_root_dir")
|
||||
|
||||
Reference in new issue
Block a user