Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer,
 
 langs = {"English": ("en", "eng_Latn"), "Romanian": ("ro", "ron_Latn"), "German": ("de", "deu_Latn"), "French": ("fr", "fra_Latn"), "Spanish": ("es", "spa_Latn"), "Italian": ("it", "ita_Latn")}
 options = list(langs.keys())
-models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "nllb-200-distilled-1.3B", "nllb-200-distilled-600M"]
+models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-distilled-600M"]
 
 @spaces.GPU
 def translate_text(input_text, sselected_language, tselected_language, model_name):
@@ -22,7 +22,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
                 model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
             except EnvironmentError as error:
                 return f"Error finding required model! Try other available language combination. Error: {error}"
-    elif model_name.startswith('nllb'):
+    elif model_name.startswith('facebook/nllb'):
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=langs[sselected_language][1], tgt_lang=langs[tselected_language][1])