Kingston Yip commited on
Commit
6f318d4
1 Parent(s): 5a3e810
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -6,14 +6,13 @@ st.title("Toxic Tweets Analyzer")
6
  image = "kanye_tweet.jpg"
7
  st.image(image, use_column_width=True)
8
 
 
 
 
9
  #form
10
  with st.form("my_form"):
11
  submitted = st.form_submit_button("Analyze")
12
  tweet = st.text_area("enter tweet here:", value="i'm nice at ping pong")
13
  if submitted:
14
- out = pipe(tweet)
15
  st.json(out)
16
- # # print out nicely
17
- # sentiment = out[0]['label']
18
- # score = out[0]['score']
19
- # print(f"{tweet} \nsentiment: {sentiment} \nscore: {score}")
 
6
  image = "kanye_tweet.jpg"
7
  st.image(image, use_column_width=True)
8
 
9
+ # create a dropdown to select the model
10
+ model = st.selectbox("Select model", ["distilbert-base-uncased-finetuned-sst-2-english", "nlptown/bert-base-multilingual-uncased-sentiment"])
11
+
12
  #form
13
  with st.form("my_form"):
14
  submitted = st.form_submit_button("Analyze")
15
  tweet = st.text_area("enter tweet here:", value="i'm nice at ping pong")
16
  if submitted:
17
+ out = pipe(tweet, model=model)
18
  st.json(out)