From ad108060ed8da22eaf31007af06b5153ae5018e8 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 6 Feb 2023 18:06:19 -0600 Subject: [PATCH] fix(api): only try to load extras when a path is provided (#112) --- api/onnx_web/convert.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/api/onnx_web/convert.py b/api/onnx_web/convert.py index e36761fe..79177fae 100644 --- a/api/onnx_web/convert.py +++ b/api/onnx_web/convert.py @@ -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