Zack
adding another emotion model
61ff67d
raw
history blame
306 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)