TiberiuCristianLeon commited on
Commit
788c01d
·
verified ·
1 Parent(s): 13ac8cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -102,7 +102,7 @@ class Translators:
102
  translated_text = error
103
  return translated_text
104
 
105
- def HelsinkiNLP_mulmul(self):
106
  try:
107
  pipe = pipeline("translation", model=self.model_name, device=self.device)
108
  iso_dict = {iso[1]: iso[3] for iso in non_empty_isos}
@@ -400,8 +400,8 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
400
  message_text = f'Translated from {s_language} to {t_language} with {model_name}'
401
  translated_text = None
402
  try:
403
- if model_name == "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul":
404
- translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP_mulmul()
405
 
406
  elif model_name == "Helsinki-NLP":
407
  translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()
 
102
  translated_text = error
103
  return translated_text
104
 
105
+ def HelsinkiNLP_mul(self):
106
  try:
107
  pipe = pipeline("translation", model=self.model_name, device=self.device)
108
  iso_dict = {iso[1]: iso[3] for iso in non_empty_isos}
 
400
  message_text = f'Translated from {s_language} to {t_language} with {model_name}'
401
  translated_text = None
402
  try:
403
+ if "-mul" in model_name.lower() or "mul-" in model_name.lower():
404
+ translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP_mul()
405
 
406
  elif model_name == "Helsinki-NLP":
407
  translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()