!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() |