llm-test / app.py
Samantha Hipple
project init
fe68765
raw
history blame
270 Bytes
import streamlit as st
from transformers import pipeline
from PIL import Image
generator = pipeline(task='text-generation', model="gpt2")
st.title("Affective Journaling Assistant")
emotion_data = st.text_input('Enter FER Results')
st.write(generator(emotion_data))