TiberiuCristianLeon commited on
Commit
6eefb6d
·
verified ·
1 Parent(s): 5b50e0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -438,13 +438,13 @@ def swap_languages(src_lang, tgt_lang):
438
  def get_info(model_name, sl, tl):
439
  if model_name == "Helsinki-NLP":
440
  response = requests.get(f'https://huggingface.co/{model_name}/opus-mt-{sl}-{tl}/raw/main/README.md').text
441
- if 'Repository not found' in response:
442
- return '# [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP")'
443
  return response
444
  if model_name == "Argos":
445
  return requests.get(f'https://huggingface.co/TiberiuCristianLeon/Argostranslate/raw/main/README.md').text
446
  if model_name == "Google":
447
- return "### Google Translate"
448
  return requests.get(f'https://huggingface.co/{model_name}/raw/main/README.md').text
449
 
450
  def create_interface():
@@ -458,7 +458,7 @@ def create_interface():
458
  swap_btn = gr.Button("Swap Languages", size="md", scale=1)
459
  swap_btn.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
460
 
461
- with gr.Row():
462
  model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {models[0]}.", value = models[0], interactive=True, scale=3)
463
  translate_btn = gr.Button("Translate", scale=1)
464
 
 
438
  def get_info(model_name, sl, tl):
439
  if model_name == "Helsinki-NLP":
440
  response = requests.get(f'https://huggingface.co/{model_name}/opus-mt-{sl}-{tl}/raw/main/README.md').text
441
+ if 'Repository not found' in response or 'Invalid username or password' in response:
442
+ return '### [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP")'
443
  return response
444
  if model_name == "Argos":
445
  return requests.get(f'https://huggingface.co/TiberiuCristianLeon/Argostranslate/raw/main/README.md').text
446
  if model_name == "Google":
447
+ return "Google Translate"
448
  return requests.get(f'https://huggingface.co/{model_name}/raw/main/README.md').text
449
 
450
  def create_interface():
 
458
  swap_btn = gr.Button("Swap Languages", size="md", scale=1)
459
  swap_btn.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
460
 
461
+ with gr.Row(equal_height=True):
462
  model_name = gr.Dropdown(choices=models, label=f"Select a model. Default is {models[0]}.", value = models[0], interactive=True, scale=3)
463
  translate_btn = gr.Button("Translate", scale=1)
464