cs670project / app.py
JasmineQiuqiu
Update app.py
b236a9b unverified
raw
history blame
No virus
1.11 kB
import streamlit as st
from transformers import pipeline
# title
st.title('Analyze Sentiment of Text!')
# Display text
st.write("You will get a sentiment analysis after you enter something.")
# show an emoji
st.write("Let's try it: :smiley:")
user_input = st.text_input('Enter something below:')
model_options = ["finiteautomata/bertweet-base-sentiment-analysis",
"nlptown/bert-base-multilingual-uncased-sentiment",
"bhadresh-savani/distilbert-base-uncased-emotion"]
selected_model = st.selectbox('Choose a model:', model_options)
def choose_model(option):
chosen_model = pipeline("sentiment-analysis", model=option)
return chosen_model
# Initialize the model
model = None
if selected_model:
model = choose_model(selected_model)
if st.button('Submit'):
if user_input:
if model:
results = model(user_input)
st.write('The detected sentiment is', results[0]['label'], "with a score of", results[0]['score'])
else:
st.write('Please select a model!')
else:
st.write('Please enter something!')