Felix Marty commited on
Commit
c9b76af
1 Parent(s): 0fead51
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -31,6 +31,8 @@ def onnx_export(token: str, model_id: str, task: str, opset: Union[int, str]) ->
31
  try:
32
  if opset == "":
33
  opset = None
 
 
34
 
35
  api = HfApi(token=token)
36
 
 
31
  try:
32
  if opset == "":
33
  opset = None
34
+ else:
35
+ opset = int(opset)
36
 
37
  api = HfApi(token=token)
38