Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| from transformers import pipeline | |
| classifier = pipeline("zero-shot-classification", | |
| model="valhalla/distilbart-mnli-12-1") | |
| input_text = st.text_area("Enter your sample") | |
| input_label = st.text_input("Enter labels", placeholder="support, help, important") | |
| labels = input_label.split(',') | |
| pred = classifier(input_text, labels, multi_class=True) | |
| st.markdown(pred) | |