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
4
5checkpoint = "Mr-Vicky-01/English-Tamil-Translator"
6tokenizer = AutoTokenizer.from_pretrained(checkpoint)
7model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
8
9def language_translator(text):
10 tokenized = tokenizer([text], return_tensors='pt')
11 out = model.generate(**tokenized, max_length=128)
12 return tokenizer.decode(out[0],skip_special_tokens=True)
13
14text_to_translate = "hardwork never fail"
15output = language_translator(text_to_translate)
16print(output)