Update app.py
Browse files
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 '
|
| 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 "
|
| 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 |
|