Update app.py
Browse files
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
|
| 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
|
| 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).
|
| 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()
|