metadata
license: apache-2.0
widget:
- text: I was born at 2003 and my sister was born at 2000.
- text: How are You?
- text: Actor Vijay is competing an 2026 election.
language:
- ta
def language_translator(text):
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
model = AutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/")
# model = AutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/")
tokenized = tokenizer([text], return_tensors='pt')
out = model.generate(**tokenized, max_length=128)
with tokenizer.as_target_tokenizer():
return tokenizer.decode(out[0],skip_special_tokens=True)
text_to_translate = "How are you?"
output = language_translator(text_to_translate)
print(output)