Update app.py

app.py

@@ -3,7 +3,10 @@ import spaces
 import torch
 from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
-langs = {"German": "de", "Romanian": "ro", "English": "en", "French": "fr", "Spanish": "es", "Italian": "it"}
+favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "French": "fr", "-----": "-----"}
+all_langs = {"German": "de", "Romanian": "ro", "English": "en", "French": "fr", "Spanish": "es", "Italian": "it", "Hungarian": "hu"}
+langs = {**favourite_langs, **all_langs}
+
 options = list(langs.keys())
 models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/mbart-large-50-many-to-many-mmt"]
 
@@ -22,16 +25,16 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
     message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
     if model_name == "Helsinki-NLP":
         try:
-            model_name_full = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
-            tokenizer = AutoTokenizer.from_pretrained(model_name_full)
-            model = model_to_cuda(AutoModelForSeq2SeqLM.from_pretrained(model_name_full))
+            model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
+            tokenizer = AutoTokenizer.from_pretrained(model_name)
+            model = model_to_cuda(AutoModelForSeq2SeqLM.from_pretrained(model_name))
         except EnvironmentError:
-            try :   
-                model_name_full = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
-                tokenizer = AutoTokenizer.from_pretrained(model_name_full)
-                model = model_to_cuda(AutoModelForSeq2SeqLM.from_pretrained(model_name_full))
+            try:   
+                model_name = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
+                tokenizer = AutoTokenizer.from_pretrained(model_name)
+                model = model_to_cuda(AutoModelForSeq2SeqLM.from_pretrained(model_name))
             except EnvironmentError as error:
-                return f"Error finding model: {model_name_full}! Try other available language combination.", error
+                return f"Error finding model: {model_name}! Try other available language combination.", error
     
     if model_name.startswith('facebook/nllb'):
         from languagecodes import nllb_language_codes
@@ -91,8 +94,8 @@ def create_interface():
         model_name = gr.Dropdown(choices=models, label="Select a model", value = models[4], interactive=True)
         translate_button = gr.Button("Translate")
 
-        translated_text = gr.Textbox(label="Translated text:", interactive=False, show_copy_button=True)
-        message_text = gr.Textbox(label="Messages:", placeholder="Display status and error messages", interactive=False)
+        translated_text = gr.Textbox(label="Translated text:", placeholder="Display field for translation", interactive=False, show_copy_button=True)
+        message_text = gr.Textbox(label="Messages:", placeholder="Display field for status and error messages", interactive=False)
 
         translate_button.click(
             translate_text, 
@@ -102,6 +105,5 @@ def create_interface():
 
     return interface
 
-# Launch the Gradio interface
 interface = create_interface()
 interface.launch()