Felix Marty commited on
Commit
3c5a0a8
1 Parent(s): 5bc03ad
Files changed (1) hide show
  1. app.py +21 -20
app.py CHANGED
@@ -107,26 +107,27 @@ Note: in case the model to convert is larger than 2 GB, it will be saved in a su
107
  """
108
 
109
  with gr.Blocks() as demo:
110
- with gr.Column(scale=50):
111
- gr.HTML(TTILE_IMAGE)
112
- gr.HTML(TITLE)
113
- gr.Markdown(DESCRIPTION)
114
-
115
- with gr.Column(scale=50):
116
- input_token = gr.Textbox(max_lines=1, label="Hugging Face token")
117
- input_model = gr.Textbox(
118
- max_lines=1,
119
- label="Model name",
120
- placeholder="textattack/distilbert-base-cased-CoLA",
121
- )
122
- input_task = gr.Textbox(
123
- value="auto",
124
- max_lines=1,
125
- label='Task (can be left to "auto", will be automatically inferred)',
126
- )
127
-
128
- btn = gr.Button("Convert to ONNX")
129
- output = gr.Markdown(label="Output")
 
130
 
131
  btn.click(
132
  fn=onnx_export, inputs=[input_token, input_model, input_task], outputs=output
 
107
  """
108
 
109
  with gr.Blocks() as demo:
110
+ with gr.Row():
111
+ with gr.Column(scale=50):
112
+ gr.HTML(TTILE_IMAGE)
113
+ gr.HTML(TITLE)
114
+ gr.Markdown(DESCRIPTION)
115
+
116
+ with gr.Column(scale=50):
117
+ input_token = gr.Textbox(max_lines=1, label="Hugging Face token")
118
+ input_model = gr.Textbox(
119
+ max_lines=1,
120
+ label="Model name",
121
+ placeholder="textattack/distilbert-base-cased-CoLA",
122
+ )
123
+ input_task = gr.Textbox(
124
+ value="auto",
125
+ max_lines=1,
126
+ label='Task (can be left to "auto", will be automatically inferred)',
127
+ )
128
+
129
+ btn = gr.Button("Convert to ONNX")
130
+ output = gr.Markdown(label="Output")
131
 
132
  btn.click(
133
  fn=onnx_export, inputs=[input_token, input_model, input_task], outputs=output