fix

app.py

@@ -1,7 +1,7 @@
 import csv
 import os
 from datetime import datetime
-from typing import Optional
+from typing import Optional, Union
 
 import gradio as gr
 from huggingface_hub import HfApi, Repository
@@ -21,7 +21,7 @@ if HF_TOKEN:
     repo = Repository(local_dir=DATADIR, clone_from=DATASET_REPO_URL, token=HF_TOKEN)
 
 
-def onnx_export(token: str, model_id: str, task: str, opset: int) -> str:
+def onnx_export(token: str, model_id: str, task: str, opset: Union[int, str]) -> str:
     if token == "" or model_id == "":
         return """
         ### Invalid input 🐞
@@ -29,6 +29,9 @@ def onnx_export(token: str, model_id: str, task: str, opset: int) -> str:
         Please fill a token and model name.
         """
     try:
+        if opset == "":
+            opset = None
+
         api = HfApi(token=token)
 
         error, commit_info = convert(api=api, model_id=model_id, task=task, opset=opset)

onnx_export.py

@@ -61,7 +61,6 @@ def convert_onnx(model_id: str, task: str, folder: str, opset: int) -> List:
                 "Could not infer the pad token id, which is needed in this case, please provide it with the --pad_token_id argument"
             )
 
-    print("opset:", opset)
     # Ensure the requested opset is sufficient
     if opset == None:
         opset = onnx_config.DEFAULT_ONNX_OPSET