saiful9379 commited on
Commit
5febc01
·
1 Parent(s): efba808

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -1
README.md CHANGED
@@ -4,4 +4,24 @@ license: mit
4
  widget:
5
  - বহুল আলোচিত দশম জাতীয় সংসদ
6
  - গাজীপুরের কালিয়াকৈর উপজেলার তেলিরচালা
7
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  widget:
5
  - বহুল আলোচিত দশম জাতীয় সংসদ
6
  - গাজীপুরের কালিয়াকৈর উপজেলার তেলিরচালা
7
+ ---
8
+
9
+ from transformers import AutoTokenizer, AutoModelForCausalLM
10
+
11
+ tokenizer = AutoTokenizer.from_pretrained("saiful9379/Bangla_GPT2")
12
+
13
+ model = AutoModelForCausalLM.from_pretrained("saiful9379/Bangla_GPT2")
14
+ text = "বহুল আলোচিত দশম জাতীয় সংসদ"
15
+ input_ids = tokenizer.encode(text, return_tensors='tf')
16
+
17
+
18
+ output = model.generate(
19
+ input_ids,
20
+ max_length=175,
21
+ num_beams=10,
22
+ temperature=0.7,
23
+ no_repeat_ngram_size=2,
24
+ num_return_sequences=5
25
+ )
26
+ predicted_text = tokenizer.decode(output[0], skip_special_tokens=True)
27
+ print(predicted_text)