apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): only try to load extras when a path is provided (#112)

This commit is contained in:
Sean Sube 2023-02-06 18:06:19 -06:00
parent 7db5472531
commit ad108060ed
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
api/onnx_web/convert.py
View File

@ -564,14 +564,15 @@ def main() -> int:
load_models(args, base_models)
for file in args.extras:
logger.info("Loading extra models from %s", file)
try:
with open(file, "r") as f:
data = loads(f.read())
logger.info("Converting extra models.")
load_models(args, data)
except Exception as err:
logger.error("Error converting extra models: %s", err)
if file is not None and file != "":
logger.info("Loading extra models from %s", file)
try:
with open(file, "r") as f:
data = loads(f.read())
logger.info("Converting extra models.")
load_models(args, data)
except Exception as err:
logger.error("Error converting extra models: %s", err)
return 0