wasmdashai commited on
Commit
6714f3e
·
verified ·
1 Parent(s): a1cc1df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -61,7 +61,7 @@ class OnnxModelConverter:
61
  dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
62
  )
63
 
64
- return self.download_file(onnx_file)
65
 
66
 
67
  def convert_only_decoder(self, model_name, token, onnx_filename):
@@ -92,7 +92,7 @@ class OnnxModelConverter:
92
  dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
93
  )
94
 
95
- return self.download_file(onnx_file)
96
 
97
  def convert_full_model(self, model_name, token, onnx_filename):
98
  """
@@ -122,7 +122,7 @@ class OnnxModelConverter:
122
  dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
123
  )
124
 
125
- return self.download_file(onnx_file)
126
  def starrt(self):
127
  with gr.Blocks() as demo:
128
  with gr.Row():
 
61
  dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
62
  )
63
 
64
+ return onnx_file
65
 
66
 
67
  def convert_only_decoder(self, model_name, token, onnx_filename):
 
92
  dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
93
  )
94
 
95
+ return onnx_file
96
 
97
  def convert_full_model(self, model_name, token, onnx_filename):
98
  """
 
122
  dynamic_axes={'input': {0: 'batch_size', 1: 'sequence_length'}, 'output': {0: 'batch_size'}}
123
  )
124
 
125
+ return onnx_file
126
  def starrt(self):
127
  with gr.Blocks() as demo:
128
  with gr.Row():