File size: 360 Bytes
d02bbc6
1e988e6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d02bbc6
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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)