| from transformers import pipeline | |
| translator_id_en = pipeline("translation", model="Helsinki-NLP/opus-mt-id-en") | |
| translator_en_id = pipeline("translation", model="Helsinki-NLP/opus-mt-en-id") | |
| def translate(text, direction): | |
| if direction == "ID → EN": | |
| return translator_id_en(text)[0]['translation_text'] | |
| else: | |
| return translator_en_id(text)[0]['translation_text'] | |