A model that translates English sentences to Tamil.
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2model_checkpoint = "aryaumesh/english-to-tamil"
3
4tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
5model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
6
7def language_translator(text):
8 tokenized = tokenizer([text], return_tensors='pt')
9 out = model.generate(**tokenized, max_length=128)
10 return tokenizer.decode(out[0],skip_special_tokens=True)
11
12text_to_translate = "This is a test case." # Sentence to translate
13output = language_translator(text_to_translate)
14print(output) # இது ஒரு சோதனை வழக்கு.