Felix Marty
commited on
Commit
•
3c5a0a8
1
Parent(s):
5bc03ad
add row
Browse files
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.
|
111 |
-
gr.
|
112 |
-
|
113 |
-
|
114 |
-
|
115 |
-
|
116 |
-
|
117 |
-
|
118 |
-
|
119 |
-
|
120 |
-
|
121 |
-
|
122 |
-
|
123 |
-
|
124 |
-
|
125 |
-
|
126 |
-
|
127 |
-
|
128 |
-
|
129 |
-
|
|
|
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
|