Spaces:
Runtime error
Runtime error
File size: 324 Bytes
f0b39b9 a01ae42 f0b39b9 a01ae42 9a080df | 1 2 3 4 5 6 7 8 9 | import streamlit as st
import transformers
from transformers import pipeline
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
sequence_to_classify = st.text_area("Enter the text here")
candidate_labels = ['travel', 'cooking', 'dancing']
classifier(sequence_to_classify, candidate_labels) |