Spaces:
Runtime error
Runtime error
import streamlit as st | |
from transformers import pipeline | |
classifier = pipeline("zero-shot-classification", | |
model="valhalla/distilbart-mnli-12-1") | |
with st.form('inputs') as form: | |
input_text = form.text_area("Enter your sample") | |
input_label = form.text_input("Enter labels", placeholder="support, help, important") | |
submit_button = form.form_submit_button(label='Submit') | |
if submit_button: | |
labels = list(l.strip() for l in input_label.split(',')) | |
pred = classifier(input_text, labels, multi_class=True) | |
st.markdown(pred) | |