Matt09Miao commited on
Commit
3fe7bd2
·
verified ·
1 Parent(s): 0d99644

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ toxic_model = pipeline("text-classification", model="Matt09Miao/GP5_tweet_toxic"
8
 
9
  # text2story
10
  def text2story(text):
11
- pipe = pipeline("text-generation", model="openai-community/gpt2")
12
  tweet_text = pipe(text)
13
  return tweet_text
14
 
 
8
 
9
  # text2story
10
  def text2story(text):
11
+ pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
12
  tweet_text = pipe(text)
13
  return tweet_text
14