Views
No views yet

1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("rudrashah/RLM-hinglish-translator")
4model = AutoModelForCausalLM.from_pretrained("rudrashah/RLM-hinglish-translator")
5
6template = "Hinglish:\n{hi_en}\n\nEnglish:\n{en}" #THIS IS MOST IMPORTANT, WITHOUT THIS IT WILL GIVE RANDOM OUTPUT
7input_text = tokenizer(template.format(hi_en="aapka name kya hai?",en=""),return_tensors="pt")
8
9output = model.generate(**input_text)
10print(tokenizer.decode(output[0]))