Spaces:
Runtime error
Runtime error
import streamlit as st | |
from transformers import pipeline | |
st.title("Text Classifier 1") | |
st.write("Input text and labels -- comma separate labels. Submit. The machine will classify the text according to one of the labels. It will also present a score for classifying that text for that label.") | |
classifier = pipeline(task="zero-shot-classification") | |
form = st.form(key='zeroclassifier_form') | |
input_text = form.text_input(label='Input Text') | |
labels = form.text_input(label='Labels -- comma separate labels') | |
submit_button = form.form_submit_button(label='Submit') | |
if submit_button: | |
labs = labels.title().split(',') | |
text = input_text.title() | |
res = classifier(text, labs) | |
st.success(res['labels'][0]) | |
st.info(res['scores'][0]) | |