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 |
|