Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3# Load the model and tokenizer
4model_name = "Meher2006/english-to-telugu-model"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
8# Function to translate text
9def translate(text):
10 inputs = tokenizer(text, return_tensors="pt")
11 outputs = model.generate(**inputs)
12 return tokenizer.decode(outputs[0], skip_special_tokens=True)
13
14# Example Usage
15sentences = [
16 "Hello, how are you?", # Output: హలో, మీరు ఎలా ఉన్నారు?
17 "What is your name?", # Output: మీ పేరు ఏమిటి?
18 "I enjoy learning new languages.", # Output: నాకు కొత్త భాషలు నేర్చుకోవడం ఇష్టం.
19 "Please help me with this task.", # Output: దయచేసి ఈ పనిలో నాకు సహాయపడండి.
20 "Thank you.", # Output: ధన్యవాదాలు.
21 "How are you doing?" # Output: నువ్వు ఎలా ఉన్నావు?
22]
23
24for sentence in sentences:
25 print(f"English: {sentence}")
26 print(f"Telugu: {translate(sentence)}")
27 print("----")