TiberiuCristianLeon commited on
Commit
0b2c6b5
·
verified ·
1 Parent(s): e14faa9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -474,7 +474,7 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
474
  def swap_languages(src_lang, tgt_lang):
475
  return tgt_lang, src_lang
476
 
477
- def get_info(model_name: str = DEFAULTS[2], sl: str = None, tl: str = None):
478
  helsinki = '### [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP")'
479
  if model_name == "Helsinki-NLP" and sl is not None and tl is not None:
480
  response = requests.get(f'https://huggingface.co/{model_name}/opus-mt-{sl}-{tl}/raw/main/README.md').text
@@ -500,14 +500,14 @@ def create_interface():
500
  swap_btn = gr.Button("Swap Languages", size="md", scale=1)
501
  swap_btn.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
502
  # with gr.Row(equal_height=True):
503
- model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {models[0]}.", value=models[0], interactive=True, scale=2)
504
  translate_btn = gr.Button("Translate", scale=1)
505
 
506
  translated_text = gr.Textbox(label="Translated text:", placeholder="Display field for translation", interactive=False, show_copy_button=True)
507
  message_text = gr.Textbox(label="Messages:", placeholder="Display field for status and error messages", interactive=False,
508
  value=f'Default translation settings: from {s_language.value} to {t_language.value} with {model_name.value}.')
509
  allmodels = gr.HTML(label="Model links:", value=', '.join([f'<a href="https://huggingface.co/{model}">{model}</a>' for model in models]))
510
- model_info = gr.Markdown(label="Model info:", value=get_info(model_name), show_copy_button=True)
511
  model_name.change(fn=get_info, inputs=[model_name, s_language, t_language], outputs=model_info, api_name=False, show_api=False)
512
 
513
  translate_btn.click(
 
474
  def swap_languages(src_lang, tgt_lang):
475
  return tgt_lang, src_lang
476
 
477
+ def get_info(model_name: str, sl: str = None, tl: str = None):
478
  helsinki = '### [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP")'
479
  if model_name == "Helsinki-NLP" and sl is not None and tl is not None:
480
  response = requests.get(f'https://huggingface.co/{model_name}/opus-mt-{sl}-{tl}/raw/main/README.md').text
 
500
  swap_btn = gr.Button("Swap Languages", size="md", scale=1)
501
  swap_btn.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
502
  # with gr.Row(equal_height=True):
503
+ model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {DEFAULTS[2]}.", value=DEFAULTS[2], interactive=True, scale=2)
504
  translate_btn = gr.Button("Translate", scale=1)
505
 
506
  translated_text = gr.Textbox(label="Translated text:", placeholder="Display field for translation", interactive=False, show_copy_button=True)
507
  message_text = gr.Textbox(label="Messages:", placeholder="Display field for status and error messages", interactive=False,
508
  value=f'Default translation settings: from {s_language.value} to {t_language.value} with {model_name.value}.')
509
  allmodels = gr.HTML(label="Model links:", value=', '.join([f'<a href="https://huggingface.co/{model}">{model}</a>' for model in models]))
510
+ model_info = gr.Markdown(label="Model info:", value=DEFAULTS[2], show_copy_button=True)
511
  model_name.change(fn=get_info, inputs=[model_name, s_language, t_language], outputs=model_info, api_name=False, show_api=False)
512
 
513
  translate_btn.click(