IsabelLoci's picture
Update app.py
4ed96bc verified
raw
history blame contribute delete
602 Bytes
import streamlit as st
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="michellejieli/emotion_text_classifier")
st.title("Emotion in Text Classifier")
st.write("Enter any sentence, and the model will predict the expressed emotion in it!")
user_input = st.text_area("Type your prompt here.", "...")
if st.button("Predict Emotion"):
result = classifier(user_input)
emotion = result[0]['label']
score = result[0]['score']
st.write(f"**Predicted Emotion:** {emotion}")
st.write(f"**Confidence Score:** {score:.2f}")