saiful9379 commited on
Commit
ca6cd0e
1 Parent(s): 2f599c1

update readme

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -6,14 +6,16 @@ widget:
6
  - গাজীপুরের কালিয়াকৈর উপজেলার তেলিরচালা
7
  ---
8
  ```py
9
- from transformers import AutoTokenizer, AutoModelForCausalLM
10
 
11
- tokenizer = AutoTokenizer.from_pretrained("saiful9379/Bangla_GPT2")
12
 
13
- model = AutoModelForCausalLM.from_pretrained("saiful9379/Bangla_GPT2", from_tf=True)
14
  text = "বহুল আলোচিত দশম জাতীয় সংসদ"
15
  input_ids = tokenizer.encode(text, return_tensors='tf')
16
 
 
 
17
 
18
  output = model.generate(
19
  input_ids,
 
6
  - গাজীপুরের কালিয়াকৈর উপজেলার তেলিরচালা
7
  ---
8
  ```py
9
+ from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
10
 
11
+ tokenizer = GPT2Tokenizer.from_pretrained("saiful9379/Bangla_GPT2")
12
 
13
+ model = TFGPT2LMHeadModel.from_pretrained("saiful9379/Bangla_GPT2")
14
  text = "বহুল আলোচিত দশম জাতীয় সংসদ"
15
  input_ids = tokenizer.encode(text, return_tensors='tf')
16
 
17
+ print(input_ids)
18
+
19
 
20
  output = model.generate(
21
  input_ids,