import streamlit as st | |
from transformers import pipeline | |
# Title and Subtitle | |
st.title('Sentiment Analysis App') | |
st.subheader('Enter text to analyze sentiment') | |
# Text input box | |
text = st.text_area('Enter your text') | |
if text: | |
# Analyze sentiment using the model | |
pipe = pipeline(model="dancingninjas/sentiment-analysis-nlp") | |
results = pipe(text) | |
# Determine sentiment label | |
sentiment_label = "Negative" if results[0]['label'] == 'LABEL_1' else "Positive" | |
# Display sentiment label and probability | |
st.subheader('Sentiment Analysis Result:') | |
st.write(f'Sentiment: {sentiment_label}') | |
st.write(f'Probability: {results[0]["score"]:.4f}') | |