Update app.py
Browse files
app.py
CHANGED
|
@@ -285,8 +285,8 @@ class Translators:
|
|
| 285 |
no_repeat_ngram_size=1,
|
| 286 |
repetition_penalty=2,
|
| 287 |
)
|
| 288 |
-
|
| 289 |
-
return
|
| 290 |
|
| 291 |
def smollm(self):
|
| 292 |
tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
|
|
|
| 285 |
no_repeat_ngram_size=1,
|
| 286 |
repetition_penalty=2,
|
| 287 |
)
|
| 288 |
+
translated_text = tokenizer.decode(results[0].hypotheses[0])
|
| 289 |
+
return translated_text
|
| 290 |
|
| 291 |
def smollm(self):
|
| 292 |
tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|