in-class-app / app.py
Zack
adding the app file
1e988e6
raw
history blame contribute delete
360 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 to me what happened to you today.")
if user_text:
emotion = pipe(user_text)
st.markdown("you are feeling like")
st.json(emotion)