TiberiuCristianLeon commited on
Commit
ba3b9f6
·
verified ·
1 Parent(s): a4d39dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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,