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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -27,9 +27,7 @@ translation_task_names = {
27
  model_names = {
28
  'T5-Base': 't5-base',
29
  'T5-Small': 't5-small',
30
- 'T5-Large': 't5-large',
31
- 'Opus-zh-en': 'Helsinki-NLP/opus-mt-zh-en',
32
- 'Opus-ru-en': 'Helsinki-NLP/opus-mt-ru-en'
33
  }
34
 
35
  # Create a dictionary to store loaded models
@@ -68,5 +66,5 @@ load_in_8bit = gr.Checkbox(label="Load model in 8bit")
68
  device = gr.Radio(['cpu', 'cuda'], label='Select device', default='cpu')
69
 
70
  iface = gr.Interface(launch, inputs=[model_dropdown, task_dropdown, text_input, load_in_8bit, device],
71
- outputs=gr.outputs.Textbox(type="text", label="Translation"))
72
  iface.launch()
 
27
  model_names = {
28
  'T5-Base': 't5-base',
29
  'T5-Small': 't5-small',
30
+ 'T5-Large': 't5-large'
 
 
31
  }
32
 
33
  # Create a dictionary to store loaded models
 
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"))
70
  iface.launch()