Update app.py

app.py

@@ -2,12 +2,14 @@ import gradio as gr
 import spaces
 from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer, AutoModelForSeq2SeqLM
 
+langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es", "Italian": "it"}
+sl = langs[sselected_language]
+tl = langs[tselected_language]
+options = ["German", "Romanian", "English", "French", "Spanish", "Italian"]
+models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large"]
+
 @spaces.GPU
 def translate_text(input_text, sselected_language, tselected_language, model_name):
-    langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es", "Italian": "it"}
-    sl = langs[sselected_language]
-    tl = langs[tselected_language]
-
     if model_name == "Helsinki-NLP":
         try:
             model_name_full = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
@@ -18,11 +20,9 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
                 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)
-            
+            except EnvironmentError as error:
+                return f"Error finding required model! Try other available language combination. Error: {error}"
+                
     else:
         tokenizer = T5Tokenizer.from_pretrained(model_name)
         model = T5ForConditionalGeneration.from_pretrained(model_name)
@@ -38,9 +38,6 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
 
     return translated_text
 
-options = ["German", "Romanian", "English", "French", "Spanish", "Italian"]
-models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large"]
-
 # Define a function to swap dropdown values
 def swap_languages(src_lang, tgt_lang):
     return tgt_lang, src_lang