Matt09Miao commited on
Commit
3eecdbe
·
verified ·
1 Parent(s): 99e6bfb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import pipeline
6
  toxic_model = pipeline("text-classification", model="Matt09Miao/GP5_tweet_toxic")
7
 
8
 
9
- text2tweet = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
10
 
11
  # text2audio
12
  def text2audio(toxic_result):
 
6
  toxic_model = pipeline("text-classification", model="Matt09Miao/GP5_tweet_toxic")
7
 
8
 
9
+ text2tweet = pipeline("text2text-generation", model="google/t5-v1_1-base")
10
 
11
  # text2audio
12
  def text2audio(toxic_result):