TiberiuCristianLeon commited on
Commit
b9fad3b
·
verified ·
1 Parent(s): 278e4fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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['pegasusenro'] or model_name == MODELS['t5enro']:
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)