Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,6 @@ MODELS = {'benro': 'BlackKakapo/opus-mt-en-ro',
|
|
| 14 |
'gemma': 'Gargaz/gemma-2b-romanian-better',
|
| 15 |
'mbartenro': 'ancebuc/mbart-translation-en-ro',
|
| 16 |
't5enro': "ancebuc/t5-translation-en-ro",
|
| 17 |
-
'pegasusenro': "ancebuc/pegasus-translation-en-ro",
|
| 18 |
'paraphrase': 'tuner007/pegasus_paraphrase'}
|
| 19 |
EMBEDDING_MODELS = {"all-MiniLM-L6-v2":384,
|
| 20 |
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2":384,
|
|
@@ -73,7 +72,7 @@ def translate(input_text: str, model_name: str = MODELS['mttcbig'], sl: str = 'e
|
|
| 73 |
# text2textgenerationpipe, translationpipe
|
| 74 |
elif model_name == MODELS['mbartenro']:
|
| 75 |
translation, message = src.Translate.Translators(model_name, sl, tl, input_text).text2textgenerationpipe()
|
| 76 |
-
elif model_name == MODELS['
|
| 77 |
translation, message = src.Translate.Translators(model_name, sl, tl, input_text).text2textgenerationpipe()
|
| 78 |
else:
|
| 79 |
translation: str = src.Translate.gemma_direct(input_text, model_name)
|
|
|
|
| 14 |
'gemma': 'Gargaz/gemma-2b-romanian-better',
|
| 15 |
'mbartenro': 'ancebuc/mbart-translation-en-ro',
|
| 16 |
't5enro': "ancebuc/t5-translation-en-ro",
|
|
|
|
| 17 |
'paraphrase': 'tuner007/pegasus_paraphrase'}
|
| 18 |
EMBEDDING_MODELS = {"all-MiniLM-L6-v2":384,
|
| 19 |
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2":384,
|
|
|
|
| 72 |
# text2textgenerationpipe, translationpipe
|
| 73 |
elif model_name == MODELS['mbartenro']:
|
| 74 |
translation, message = src.Translate.Translators(model_name, sl, tl, input_text).text2textgenerationpipe()
|
| 75 |
+
elif model_name == MODELS['t5enro']:
|
| 76 |
translation, message = src.Translate.Translators(model_name, sl, tl, input_text).text2textgenerationpipe()
|
| 77 |
else:
|
| 78 |
translation: str = src.Translate.gemma_direct(input_text, model_name)
|