Update app.py

app.py

@@ -26,9 +26,9 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
         model_name_full = "facebook/nllb-200-distilled-1.3B"
         tokenizer = AutoTokenizer.from_pretrained(model_name_full)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
-        translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt_lang='ron_Latn')
-        translated_text = translator(input_text, max_length<=360)
-        return translated_text
+        translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt_lang='ron_Latn', max_length<=360)
+        translated_text = translator(input_text)
+        return translated_text[0][‘translation_text’]
     else:
         tokenizer = T5Tokenizer.from_pretrained(model_name)
         model = T5ForConditionalGeneration.from_pretrained(model_name, device_map="auto")