Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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
|