dancingninjas's picture
Update app.py
f90c3a5
raw
history blame
817 Bytes
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 = "Positive" if results[0]['label'] == 'LABEL_1' else "Negative"
# Display sentiment label and probability
st.subheader('Sentiment Analysis Result:')
st.write(f'Sentiment: {sentiment_label}')
st.write(f'Probability: {results[0]["score"]:.4f}')
# Visualize the sentiment with an emoji
if sentiment_label == 'Positive':
st.write('πŸ˜ƒ')
else:
st.write('πŸ˜”')