madani's picture
Update test.py
9e6c31c
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:")
st.json(classifier(text))