File size: 508 Bytes
e89435c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
model_name="en2fa"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
def translate(user_text):
inputs = tokenizer.encode(user_text, return_tensors="pt")
translated = model.generate(inputs)
translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
print(translated_text)
return translated_text
translate("I am calling from tax office.")
|