Update app.py
Browse files
app.py
CHANGED
@@ -40,6 +40,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
|
|
40 |
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|
41 |
output_ids = model.generate(input_ids, max_length=512)
|
42 |
translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
|
|
43 |
print(f'Translating from {sselected_language} to {tselected_language} with {model_name}:', f'{input_text} = {translated_text}', sep='\n')
|
44 |
return translated_text
|
45 |
|
@@ -60,10 +61,11 @@ def create_interface():
|
|
60 |
swap_button = gr.Button("Swap Languages")
|
61 |
swap_button.click(fn=swap_languages, inputs=[sselected_language, tselected_language], outputs=[sselected_language, tselected_language])
|
62 |
|
63 |
-
model_name = gr.Dropdown(choices=models, label="Select a model")
|
64 |
translate_button = gr.Button("Translate")
|
65 |
|
66 |
translated_text = gr.Textbox(label="Translated text:", interactive=False)
|
|
|
67 |
|
68 |
translate_button.click(
|
69 |
translate_text,
|
|
|
40 |
input_ids = tokenizer.encode(prompt, return_tensors="pt")
|
41 |
output_ids = model.generate(input_ids, max_length=512)
|
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 |
|
|
|
61 |
swap_button = gr.Button("Swap Languages")
|
62 |
swap_button.click(fn=swap_languages, inputs=[sselected_language, tselected_language], outputs=[sselected_language, tselected_language])
|
63 |
|
64 |
+
model_name = gr.Dropdown(choices=models, label="Select a model", value = models[4], interactive=True)
|
65 |
translate_button = gr.Button("Translate")
|
66 |
|
67 |
translated_text = gr.Textbox(label="Translated text:", interactive=False)
|
68 |
+
message_text = gr.Textbox(label="Message:", interactive=False)
|
69 |
|
70 |
translate_button.click(
|
71 |
translate_text,
|