Matt09Miao commited on
Commit
d0d54d8
·
verified ·
1 Parent(s): a43888b

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="pranavpsv/genre-story-generator-v2")
12
  tweet_text_text = pipe(text)[0]['generated_text']
13
  return tweet_text
14
 
 
8
 
9
  # text2story
10
  def text2story(text):
11
+ pipe = pipeline("text-generation", model="distilbert/distilgpt2")
12
  tweet_text_text = pipe(text)[0]['generated_text']
13
  return tweet_text
14