Leo Liu commited on
Commit
1208e07
·
verified ·
1 Parent(s): f77ca14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def text2story(text):
18
  f"Output only the body of the story, do not include other content."
19
  )
20
 
21
- story_text = pipeline("text-generation", model="distilgpt2",
22
  max_length=100,
23
  do_sample=True,
24
  top_p=0.95,
 
18
  f"Output only the body of the story, do not include other content."
19
  )
20
 
21
+ story_text = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Distill-Qwen-7B",
22
  max_length=100,
23
  do_sample=True,
24
  top_p=0.95,