Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Felix Marty
commited on
Commit
•
c9b76af
1
Parent(s):
0fead51
int opset
Browse files
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 |
|