First-app / app.py
Khaled21's picture
Emotions
66b4a7c
raw
history blame
332 Bytes
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 your feelings here")
if user_text:
emotion = pipe(user_text)
st.markdown("You are feeling like ")
st.json(emotion)