diff --git a/api/onnx_web/convert/utils.py b/api/onnx_web/convert/utils.py index 42cb7370..61b6547f 100644 --- a/api/onnx_web/convert/utils.py +++ b/api/onnx_web/convert/utils.py @@ -192,7 +192,7 @@ def check_ext(name: str, exts: List[str]) -> Tuple[bool, str]: _name, ext = path.splitext(name) ext = ext.strip(".") - return (name in exts, ext) + return (ext in exts, ext) def source_format(model: Dict) -> Optional[str]: