Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3model_name = "hima06varshini/english-to-telugu-translation"
4token = "YOUR_ACCESS_TOKEN" # Replace with your Hugging Face token if required
5
6# Load Model & Tokenizer
7model = AutoModelForSeq2SeqLM.from_pretrained(model_name, token=token)
8tokenizer = AutoTokenizer.from_pretrained(model_name, token=token)
9
10def translate(text):
11 inputs = tokenizer(text, return_tensors="pt")
12 outputs = model.generate(**inputs)
13 return tokenizer.decode(outputs[0], skip_special_tokens=True)
14
15# Example Translation
16text = "Hello, how are you?"
17print(translate(text))