wasmdashai commited on
Commit
ddf2b1e
·
verified ·
1 Parent(s): b7e36b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ class OnnxModelConverter:
12
  if not os.path.exists(file_path):
13
  with open(file_path, "w") as file:
14
  file.write("This is a test ONNX model.")
15
- return gr.File(file_path, label="Download ONNX File")
16
 
17
 
18
  def convert(self, model_name, token, onnx_filename, conversion_type):
@@ -139,7 +139,7 @@ class OnnxModelConverter:
139
 
140
  btn=gr.Button("convert")
141
  label=gr.Label("return name model onxx")
142
- btn.click(self.convert,[text_n_model,text_n_token,text_n_onxx,choice],[label])
143
  #choice.change(fn=function_change, inputs=choice, outputs=label)
144
  return demo
145
  c=OnnxModelConverter()
 
12
  if not os.path.exists(file_path):
13
  with open(file_path, "w") as file:
14
  file.write("This is a test ONNX model.")
15
+ return file_path
16
 
17
 
18
  def convert(self, model_name, token, onnx_filename, conversion_type):
 
139
 
140
  btn=gr.Button("convert")
141
  label=gr.Label("return name model onxx")
142
+ btn.click(self.convert,[text_n_model,text_n_token,text_n_onxx,choice],[file_path])
143
  #choice.change(fn=function_change, inputs=choice, outputs=label)
144
  return demo
145
  c=OnnxModelConverter()