Update utils/translator.py
Browse files- utils/translator.py +7 -2
utils/translator.py
CHANGED
|
@@ -1,7 +1,12 @@
|
|
|
|
|
| 1 |
from transformers import pipeline
|
| 2 |
-
from config import HUGGINGFACE_API_KEY
|
| 3 |
|
| 4 |
def translate_text(text):
|
| 5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
result = translator(text, max_length=400)
|
| 7 |
return result[0]['translation_text']
|
|
|
|
| 1 |
+
import os
|
| 2 |
from transformers import pipeline
|
|
|
|
| 3 |
|
| 4 |
def translate_text(text):
|
| 5 |
+
token = os.getenv("HF_TOKEN")
|
| 6 |
+
translator = pipeline(
|
| 7 |
+
"translation",
|
| 8 |
+
model="Helsinki-NLP/opus-mt-tr-en",
|
| 9 |
+
use_auth_token=token
|
| 10 |
+
)
|
| 11 |
result = translator(text, max_length=400)
|
| 12 |
return result[0]['translation_text']
|