Spaces:

lpalum
/

tweet_classifier

Running

lpalum commited on 4 days ago

Commit

8edbe7c

verified ·

1 Parent(s): 665a972

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 # Load a tweet classification model from Hugging Face
-classifier = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment")
 def classify_tweet(tweet):
     result = classifier(tweet)[0]

 import gradio as gr
 from transformers import pipeline
 # Load a tweet classification model from Hugging Face
+classifier = pipeline("text-classification", model="finiteautomata/bertweet-base-sentiment-analysis")
 def classify_tweet(tweet):
     result = classifier(tweet)[0]