import streamlit as st
from transformers import pipeline

# Title and Subtitle
st.title('Sentiment Analysis App')


pipe=pipeline(model="dancingninjas/sentiment-model")
text=st.text_area('Enter your text')

if text:
    out = pipe(text)

    # Convert the model's output (0 or 1) to descriptive labels
    sentiment_label = "Negative" if out[0]['label'] == 'LABEL_1' else "Positive"

    # Display the sentiment label and probability

    st.write(f'Sentiment: {sentiment_label}')
    st.write(f'Probability: {out[0]["score"]:.4f}')