wh1223 commited on
Commit
3fd8e5c
·
verified ·
1 Parent(s): 08c21c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def text2story(text):
14
  The story must contain:
15
  - Beginning, middle, end
16
  - Character development"""
17
- text_to_story_model= pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
18
  story_text = text_to_story_model(
19
  messages,
20
  max_length=200,
 
14
  The story must contain:
15
  - Beginning, middle, end
16
  - Character development"""
17
+ text_to_story_model= pipeline("text-generation", model="openai-community/gpt2")
18
  story_text = text_to_story_model(
19
  messages,
20
  max_length=200,