from transformers import pipeline | |
import streamlit as st | |
classifier = pipeline("zero-shot-classification", | |
model="facebook/bart-large-mnli") | |
text = st.text_area('Enter the text') | |
candidate_labels = ['complex', 'ambiguity', 'conflicting', 'tricky'] | |
classifier(text, candidate_labels) |