import streamlit as st
from transformers import pipeline


classifier = pipeline(model="distilbert-base-uncased-finetuned-sst-2-english")

if uinput := st.chat_input("Classify: "):
    ans = classifier(uinput)
    with st.chat_message("User"):
        st.write(ans)