haouarin commited on
Commit
7586b9d
1 Parent(s): 164d84e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def greet(prompt="أنا", maxTokens=50):
13
  text = prompt +" [MASK]"
14
  for i in range(int(maxTokens)):
15
  result = fill_mask(text)
16
- r = result[random.randint(0, 4)]["sequence"]
17
  if r not in generated_text_set:
18
  print(r)
19
  text = r + " [MASK]"
 
13
  text = prompt +" [MASK]"
14
  for i in range(int(maxTokens)):
15
  result = fill_mask(text)
16
+ r = random.choice(result)["sequence"]
17
  if r not in generated_text_set:
18
  print(r)
19
  text = r + " [MASK]"