Jessiesj commited on
Commit
4a6d038
·
verified ·
1 Parent(s): 151a3dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def text2story(text):
17
  temperature=0.9, # 控制随机性(0-1,越大越随机)
18
  top_k=50, # 限制候选词数量
19
  top_p=0.95, # 核采样参数
20
- repetition_penalty=1.2))
21
  # story_text = "" # to be completed
22
  story_text = text_to_story(text)[0]["generated_text"]
23
  return story_text
 
17
  temperature=0.9, # 控制随机性(0-1,越大越随机)
18
  top_k=50, # 限制候选词数量
19
  top_p=0.95, # 核采样参数
20
+ repetition_penalty=1.2)
21
  # story_text = "" # to be completed
22
  story_text = text_to_story(text)[0]["generated_text"]
23
  return story_text