Views
No views yet
transformers library or you can use the model via the Hugging Face inference API.transformers library installed. You can install it via pip:pip install transformers1# Load model directly
2import torch
3from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4checkpoint = "TabassumMehattar/sawit_En_to_ta_conversion"
5tokenizer = AutoTokenizer.from_pretrained(checkpoint)
6model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
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)
11text_to_translate = "hardwork never fail"
12output = language_translator(text_to_translate)
13print(output)