namuisam commited on
Commit
86b1ea3
·
verified ·
1 Parent(s): 5bb88ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ def text2story(text):
15
  min_length=50,
16
  max_length=100,
17
  do_sample=True,
18
- early_stopping=True
 
19
  )[0]["generated_text"]
20
  return story_text
21
 
 
15
  min_length=50,
16
  max_length=100,
17
  do_sample=True,
18
+ early_stopping=True,
19
+ top_p=0.4
20
  )[0]["generated_text"]
21
  return story_text
22