wasmdashai commited on
Commit
93d0c27
·
verified ·
1 Parent(s): 65fe4e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", 1: "1*192*sequence_length"},
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