Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer,
 
 @spaces.GPU
 def translate_text(input_text, sselected_language, tselected_language, model_name):
-    langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es"}
+    langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es", , "Italian": "it"}
     sl = langs[sselected_language]
     tl = langs[tselected_language]
 
@@ -14,9 +14,15 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
             tokenizer = AutoTokenizer.from_pretrained(model_name_full)
             model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
         except EnvironmentError:
-            model_name_full = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
-            tokenizer = AutoTokenizer.from_pretrained(model_name_full)
-            model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
+            try :   
+                model_name_full = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
+                tokenizer = AutoTokenizer.from_pretrained(model_name_full)
+                model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
+            except EnvironmentError:
+                model_name_full = f"Helsinki-NLP/opus-tatoeba-{tl}-{sl}"
+                tokenizer = AutoTokenizer.from_pretrained(model_name_full)
+                model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
+            
     else:
         tokenizer = T5Tokenizer.from_pretrained(model_name)
         model = T5ForConditionalGeneration.from_pretrained(model_name)
@@ -41,11 +47,10 @@ def swap_languages(src_lang, tgt_lang):
 
 def create_interface():
     with gr.Blocks() as interface:
-        gr.Markdown("## Text Machine Translation")
+        gr.Markdown("## Machine Text Translation")
 
         with gr.Row():
             input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here...")
-        # Create the interface with gr.Blocks() as demo: with gr.Row(): sselected_language = gr.Dropdown(choices=options, value="German", label="Source language") tselected_language = gr.Dropdown(choices=options, value="Romanian", label="Target language") swap_button = gr.Button("Swap Languages") # Set up the button to call the swap function
         
         with gr.Row():
             sselected_language = gr.Dropdown(choices=options, value="German", label="Source language")