Spaces:
Build error
Build error
| import streamlit as st | |
| from transformers import pipeline | |
| classifier = pipeline("text-classification", model="isom5240/bert-base-uncased-emotion") | |
| st.title("Text Classification for you") | |
| st.write("Classification for 6 emotions: sadness, joy, love, anger, fear, surprice") | |
| text = st.text_area("Enter the text to classify", "") | |
| if st.button("Classify"): | |
| results = classifier(text)[0] | |
| max_score = float("-inf") | |
| max_label = "" | |
| for result in results: | |
| if result['score'] > max_score: | |
| max_score = result['score'] | |
| max_label = result['label'] | |
| st.write("Text:", text) | |
| st.write("Label:", max_label) | |
| st.write("Score:", max_score) | |