80a4d3e
2e02e9b
d37d096
80a4d3e
|
|
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)
|
