Spaces:
Runtime error
Runtime error
Mediocreatmybest
commited on
Commit
•
a760944
1
Parent(s):
365f5c4
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,8 @@ model_dropdown = gr.Dropdown(choices=list(model_names.keys()), label='Select Mod
|
|
63 |
task_dropdown = gr.Dropdown(choices=list(translation_task_names.keys()), label='Select Translation Task')
|
64 |
text_input = gr.Textbox(label="Input Text") # Single line text input
|
65 |
load_in_8bit = gr.Checkbox(label="Load model in 8bit")
|
66 |
-
|
|
|
67 |
|
68 |
iface = gr.Interface(launch, inputs=[model_dropdown, task_dropdown, text_input, load_in_8bit, device],
|
69 |
outputs=gr.Textbox(type="text", label="Translation"))
|
|
|
63 |
task_dropdown = gr.Dropdown(choices=list(translation_task_names.keys()), label='Select Translation Task')
|
64 |
text_input = gr.Textbox(label="Input Text") # Single line text input
|
65 |
load_in_8bit = gr.Checkbox(label="Load model in 8bit")
|
66 |
+
# https://www.gradio.app/docs/radio
|
67 |
+
device = gr.Radio(['cpu', 'cuda'], label='Select device', value='cpu')
|
68 |
|
69 |
iface = gr.Interface(launch, inputs=[model_dropdown, task_dropdown, text_input, load_in_8bit, device],
|
70 |
outputs=gr.Textbox(type="text", label="Translation"))
|