Mediocreatmybest commited on
Commit
a760944
1 Parent(s): 365f5c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- device = gr.Radio(['cpu', 'cuda'], label='Select device', default='cpu')
 
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"))