Update app.py

app.py

@@ -12,6 +12,9 @@ MODELS = {'benro': 'BlackKakapo/opus-mt-en-ro',
           'broen': 'BlackKakapo/opus-mt-ro-en',
           'mttcbig': 'Helsinki-NLP/opus-mt-tc-big-en-ro',
           'gemma': 'Gargaz/gemma-2b-romanian-better',
+          'mbartenro': 'ancebuc/mbart-translation-en-ro',
+          't5enro': "ancebuc/t5-translation-en-ro"),
+          'pegasusenro': "ancebuc/ancebuc/pegasus-translation-en-ro"),
           'paraphrase': 'tuner007/pegasus_paraphrase'}
 EMBEDDING_MODELS = {"all-MiniLM-L6-v2":384,
 "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2":384,
@@ -67,6 +70,11 @@ def translate(input_text: str, model_name: str = MODELS['mttcbig'], sl: str = 'e
         translation, model_name = src.Translate.paraphraseTranslateMethod(input_text, model_name)
     elif 'Helsinki-NLP' in model_name:
         translation, message = src.Translate.Translators(model_name, sl, tl, input_text).HelsinkiNLP_mulroa()
+     # text2textgenerationpipe, translationpipe
+    elif model_name == MODELS['mbartenro']:
+        translation, message = src.Translate.Translators(model_name, sl, tl, input_text).text2textgenerationpipe()
+    elif model_name == MODELS['pegasusenro'] or model_name == MODELS['t5enro']:
+        translation, message = src.Translate.Translators(model_name, sl, tl, input_text).text2textgenerationpipe()
     else:
         translation: str = src.Translate.gemma_direct(input_text, model_name)
     return {"input_text": input_text, "translation": translation, "model_name": model_name, "message": message}