Spaces:
Runtime error
Runtime error
Mediocreatmybest
commited on
Commit
·
365f5c4
1
Parent(s):
c9549b9
Update app.py
Browse files
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.
|
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()
|