Views
No views yet
gopi30/english-to-tamil-stage1M2M100ForConditionalGenerationfacebook/m2m100_418M.transformers library installed.pip install transformers1from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
2
3model_path = "gopi30/english-to-tamil-stage1"
4
5model = M2M100ForConditionalGeneration.from_pretrained(model_path)
6tokenizer = M2M100Tokenizer.from_pretrained(model_path)
7
8def translate_en_to_ta(text):
9 tokenizer.src_lang = "en"
10 encoded = tokenizer(text, return_tensors="pt")
11 generated = model.generate(**encoded, forced_bos_token_id=tokenizer.get_lang_id("ta"))
12 return tokenizer.decode(generated[0], skip_special_tokens=True)
13
14# Example
15print(translate_en_to_ta("Hello!"))
16