1from transformers import TFAutoModelForSeq2SeqLM, AutoTokenizer
2from transformers import pipeline
3
4# Load the saved model and tokenizer
5loaded_model = TFAutoModelForSeq2SeqLM.from_pretrained("ananya03/Kannada_model")
6loaded_tokenizer = AutoTokenizer.from_pretrained("ananya03/Kannada_model")
7
8# Step 9: Translate New Sentences
9def translate_text(text):
10 input_text = ">>kan<< " + text
11 inputs = loaded_tokenizer(input_text, return_tensors="tf", padding=True, truncation=True)
12 outputs = loaded_model.generate(inputs["input_ids"], num_beams=5, max_length=128, early_stopping=True)
13 translated_text = loaded_tokenizer.decode(outputs[0], skip_special_tokens=True)
14 return translated_text
15
16# Example usage
17english_text = "Good morning"
18translated_text=translate_text(english_text)
19print("Translated text:", translated_text)
