Spaces:
Sleeping
Sleeping
import streamlit as st | |
from transformers import pipeline | |
model_name = 'SamLowe/roberta-base-go_emotions' | |
pipe = pipeline( | |
task='text-classification', | |
model=model_name) | |
user_text = st.text_area("Explain to me what happened to you today.") | |
if user_text: | |
emotion = pipe(user_text) | |
st.markdown("you are feeling like") | |
st.json(emotion) | |