Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -78,7 +78,8 @@ class OnnxModelConverter:
|
|
78 |
opset_version=11,
|
79 |
input_names=['input'],
|
80 |
output_names=['output'],
|
81 |
-
dynamic_axes={
|
|
|
82 |
)
|
83 |
|
84 |
return self.download_file(onnx_file)
|
|
|
78 |
opset_version=11,
|
79 |
input_names=['input'],
|
80 |
output_names=['output'],
|
81 |
+
dynamic_axes={"input": {0: "batch_size", 1: "sequence_length"},
|
82 |
+
"output": {0: "batch_size", 1: "sequence_length"}})
|
83 |
)
|
84 |
|
85 |
return self.download_file(onnx_file)
|