wasmdashai commited on
Commit
3339a54
·
verified ·
1 Parent(s): 3442992

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -78,7 +78,8 @@ class OnnxModelConverter:
78
  opset_version=11,
79
  input_names=['input'],
80
  output_names=['output'],
81
- dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
 
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)