Update app.py

app.py

@@ -23,9 +23,8 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
             except EnvironmentError as error:
                 return f"Error finding required model! Try other available language combination. Error: {error}"
     elif model_name.startswith('nllb'):
-        model_name_full = "facebook/nllb-200-distilled-1.3B"
-        tokenizer = AutoTokenizer.from_pretrained(model_name_full)
-        model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
+        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])
         translated_text = translator(input_text)
         return translated_text[0]['translation_text']
@@ -41,7 +40,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
     input_ids = tokenizer.encode(prompt, return_tensors="pt")
     output_ids = model.generate(input_ids, max_length=512)
     translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
-
+    print(f'Translating from {sselected_language} to {tselected_language} with {model_name}:', f'{input_text} =  {translated_text}' sep='\n')
     return translated_text
 
 # Define a function to swap dropdown values