Spaces:
Runtime error
Runtime error
File size: 571 Bytes
1a80035 8f8b328 79b6ab8 1a80035 79b6ab8 8f8b328 79b6ab8 8f8b328 79b6ab8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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)
|