TiberiuCristianLeon commited on
Commit
e37b21c
·
verified ·
1 Parent(s): baa03e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -211,12 +211,12 @@ class Translators:
211
  except Exception as error:
212
  return f"Error translating with model: {self.model_name}! Try other available language combination or model.", error
213
 
214
- def salamandraTApipe(self):
215
  pipe = pipeline("text-generation", model=self.model_name)
216
  messages = [f"Translate the following text from {self.sl} into {self.tl}.\n{self.sl}: {self.input_text} \n{self.tl}:"]
217
  return pipe(messages)[0]["generated_text"]
218
 
219
- def salamandraTA(self):
220
  # from datetime import datetime
221
  text = f"Translate the following text from {self.sl} into {self.tl}.\n{self.sl}: {self.input_text} \n{self.tl}:"
222
  tokenizer = AutoTokenizer.from_pretrained(self.model_name)
@@ -609,7 +609,7 @@ def translate_text(model_name: str, s_language: str, t_language: str, input_text
609
  translated_text = Translators(model_name, sl, tl, input_text).google()
610
 
611
  elif "salamandra" in model_name.lower():
612
- translated_text = Translators(model_name, s_language, t_language, input_text).salamandraTApipe()
613
 
614
  elif "m2m" in model_name.lower():
615
  translated_text = Translators(model_name, sl, tl, input_text).mtom()
 
211
  except Exception as error:
212
  return f"Error translating with model: {self.model_name}! Try other available language combination or model.", error
213
 
214
+ def salamandratapipe(self):
215
  pipe = pipeline("text-generation", model=self.model_name)
216
  messages = [f"Translate the following text from {self.sl} into {self.tl}.\n{self.sl}: {self.input_text} \n{self.tl}:"]
217
  return pipe(messages)[0]["generated_text"]
218
 
219
+ def salamandrata(self):
220
  # from datetime import datetime
221
  text = f"Translate the following text from {self.sl} into {self.tl}.\n{self.sl}: {self.input_text} \n{self.tl}:"
222
  tokenizer = AutoTokenizer.from_pretrained(self.model_name)
 
609
  translated_text = Translators(model_name, sl, tl, input_text).google()
610
 
611
  elif "salamandra" in model_name.lower():
612
+ translated_text = Translators(model_name, s_language, t_language, input_text).salamandrata()
613
 
614
  elif "m2m" in model_name.lower():
615
  translated_text = Translators(model_name, sl, tl, input_text).mtom()