add base url for OpenAI (#166)

This commit is contained in:
KevinHuSh
2024-03-28 19:15:16 +08:00
committed by GitHub
parent be2b904daf
commit 38e5737067
8 changed files with 39 additions and 27 deletions

View File

@ -43,6 +43,8 @@ class Recognizer(object):
if not os.path.exists(model_file_path):
model_dir = snapshot_download(repo_id="InfiniFlow/deepdoc")
model_file_path = os.path.join(model_dir, task_name + ".onnx")
else:
model_file_path = os.path.join(model_dir, task_name + ".onnx")
if not os.path.exists(model_file_path):
raise ValueError("not find model file path {}".format(