Views
No views yet
1# Load model directly from transformers library
2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
4tokenizer = AutoTokenizer.from_pretrained("Deepakvictor/tan-ta")
5model = AutoModelForSeq2SeqLM.from_pretrained("Deepakvictor/tan-ta")
6
7#pass the input
8inp = tokenizer("Thalaivaru nirantharam",return_tensors="pt")
9out= model.generate(**inp)
10tokenizer.batch_decode(out,skip_special_tokens=True)
11#['தலைவரு நிரந்தரம்']