haouarin commited on
Commit
b8e9681
·
1 Parent(s): 23178ee

Create run_lm.py

Browse files
Files changed (1) hide show
  1. run_lm.py +17 -0
run_lm.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ fill_mask = pipeline(
3
+ "fill-mask",
4
+ model="alger-ia/dziribert",
5
+ tokenizer="alger-ia/dziribert"
6
+ )
7
+ generated_text_set = set()
8
+ text= "هذا [MASK]"
9
+ for i in range(50):
10
+ result = fill_mask(text)
11
+ r = result[random.randint(0, 4)]["sequence"]
12
+ if r not in generated_text_set:
13
+ print(r)
14
+ text = r + " [MASK]"
15
+ generated_text_set.add(r)
16
+ else:
17
+ continue