Zack
adding another emotion model
333b7cb
raw
history blame contribute delete
291 Bytes
import streamlit as st
from transformers import pipeline
#pipe = pipeline('sentiment-analysis')
pipe = pipeline(task="text-classification", model="SamLowe/roberta-base-go_emotions", top_k=None)
text = st.text_area('enter your feelings here')
if text:
out = pipe(text)
st.json(out)