Views
No views yet
1from transformers import GPT2LMHeadModel, GPT2TokenizerFast
2
3# Load tokenizer and model from Hugging Face
4tokenizer = GPT2TokenizerFast.from_pretrained("rejauldu/bengali-gpt2-tokenizer")
5model = GPT2LMHeadModel.from_pretrained("rejauldu/bengali-gpt2")
6
7# Generate text
8inputs = tokenizer("বাংলায় স্বাগত", return_tensors="pt")
9outputs = model.generate(**inputs, max_length=50)
10print(tokenizer.decode(outputs[0]))