Spaces:
Sleeping
Sleeping
wasmdashai
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -78,7 +78,7 @@ class OnnxModelConverter:
|
|
78 |
opset_version=11,
|
79 |
input_names=['input'],
|
80 |
output_names=['output'],
|
81 |
-
dynamic_axes={"input": {0: "batch_size",
|
82 |
"output": {0: "batch_size", 1: "sequence_length"}}
|
83 |
)
|
84 |
|
|
|
78 |
opset_version=11,
|
79 |
input_names=['input'],
|
80 |
output_names=['output'],
|
81 |
+
dynamic_axes={"input": {0: "batch_size", 2: "seq_len"},
|
82 |
"output": {0: "batch_size", 1: "sequence_length"}}
|
83 |
)
|
84 |
|