# Use a pipeline as a high-level helper

import streamlit as st
from transformers import pipeline

classifier= pipeline("text-classification", model="sa-rehman/stress_classifier")

input=st.text_area("Enter your text")

if st.button("Generate"):
    output=classifier(input)
    st.success(output)