Spaces:
Running
Running
import streamlit as st | |
from transformers import pipeline | |
# Load the models | |
pipe_5_star = pipeline("text-classification", model="nlptown/bert-base-multilingual-uncased-sentiment") | |
pipe_pos_neg = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english") | |
# Streamlit app | |
st.title("Sentiment Analysis") | |
# Dropdown menu for model selection | |
model_choice = st.selectbox("Choose a model:", ["5-Star Sentiment", "Positive/Negative Sentiment"]) | |
# Input text | |
user_message = st.text_area("Enter your text:", "") | |
if st.button("Analyze"): | |
if user_message: | |
if model_choice == "5-Star Sentiment": | |
result = pipe_5_star(user_message) | |
st.write("Sentiment Analysis (5-Star):") | |
st.write(result[0]['label']) | |
elif model_choice == "Positive/Negative Sentiment": | |
result = pipe_pos_neg(user_message) | |
st.write("Sentiment Analysis (Positive/Negative):") | |
st.write(result[0]['label']) | |
else: | |
st.error("Please enter a message before analyzing.") | |