Update app.py
Browse files
app.py
CHANGED
|
@@ -42,7 +42,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
|
|
| 42 |
translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
| 43 |
message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
|
| 44 |
print(f'Translating from {sselected_language} to {tselected_language} with {model_name}:', f'{input_text} = {translated_text}', sep='\n')
|
| 45 |
-
return translated_text
|
| 46 |
|
| 47 |
# Define a function to swap dropdown values
|
| 48 |
def swap_languages(src_lang, tgt_lang):
|
|
@@ -70,7 +70,7 @@ def create_interface():
|
|
| 70 |
translate_button.click(
|
| 71 |
translate_text,
|
| 72 |
inputs=[input_text, sselected_language, tselected_language, model_name],
|
| 73 |
-
outputs=translated_text
|
| 74 |
)
|
| 75 |
|
| 76 |
return interface
|
|
|
|
| 42 |
translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
| 43 |
message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
|
| 44 |
print(f'Translating from {sselected_language} to {tselected_language} with {model_name}:', f'{input_text} = {translated_text}', sep='\n')
|
| 45 |
+
return translated_text, message_text
|
| 46 |
|
| 47 |
# Define a function to swap dropdown values
|
| 48 |
def swap_languages(src_lang, tgt_lang):
|
|
|
|
| 70 |
translate_button.click(
|
| 71 |
translate_text,
|
| 72 |
inputs=[input_text, sselected_language, tselected_language, model_name],
|
| 73 |
+
outputs=[translated_text, message_text]
|
| 74 |
)
|
| 75 |
|
| 76 |
return interface
|