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)