Spaces:
Running
Running
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!') | |