space1 / app.py
madani's picture
Update app.py
2fb65f7
import streamlit as st
from transformers import pipeline
model="distilbert-base-uncased-finetuned-sst-2-english"
classifier = pipeline("sentiment-analysis", model=model)
text = st.text_input("Input a text:")
if st.button("OK"):
st.write(classifier(text))