emotion / app.py
HimanshuA's picture
Create app.py
917cad2
import streamlit as st
from transformers import pipeline
pipe = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-emotion")
text = st.text_area('Enter some text: ')
if text:
out = pipe(text)
st.json(out)