sample / app.py
Suratsada's picture
Update app.py
567a3ac
raw
history blame
526 Bytes
!python -m pip install transformers --quiet
from transformers import pipeline
classifier = pipeline("sentiment-analysis",
model="distilbert-base-uncased-finetuned-sst-2-english")
def main():
st.title("classifier")
with st.form("questionaire"):
X = st.text_input("")
clicked = st.form_submit_button("Predict")
if clicked:
result=classifier(X)
st.success(result)
if __name__ == "__main__":
main()