Bangla_GPT2 / README.md
saiful9379's picture
Update README.md
5febc01
|
raw
history blame
831 Bytes
metadata
license: mit
widget:
  - বহুল আলোচিত দশম জাতীয় সংসদ
  - গাজীপুরের কালিয়াকৈর উপজেলার তেলিরচালা

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("saiful9379/Bangla_GPT2")

model = AutoModelForCausalLM.from_pretrained("saiful9379/Bangla_GPT2") text = "বহুল আলোচিত দশম জাতীয় সংসদ" input_ids = tokenizer.encode(text, return_tensors='tf')

output = model.generate( input_ids, max_length=175, num_beams=10, temperature=0.7, no_repeat_ngram_size=2, num_return_sequences=5 ) predicted_text = tokenizer.decode(output[0], skip_special_tokens=True) print(predicted_text)