File size: 619 Bytes
7f81307
 
7179214
7f81307
 
5a3e810
 
7179214
6f318d4
 
 
3e1c304
 
1ea0270
a222015
3e1c304
6f318d4
5a3e810
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import streamlit as st
from transformers import pipeline

pipe = pipeline(task="sentiment-analysis")
st.title("Toxic Tweets Analyzer")
image = "kanye_tweet.jpg"
st.image(image, use_column_width=True)

# create a dropdown to select the model
model = st.selectbox("Select model", ["distilbert-base-uncased-finetuned-sst-2-english", "nlptown/bert-base-multilingual-uncased-sentiment"])

#form
with st.form("my_form"):
    submitted = st.form_submit_button("Analyze")
    tweet = st.text_area("enter tweet here:", value="i'm nice at ping pong")
    if submitted:
        out = pipe(tweet, model=model)
        st.json(out)