emre-kocyigit's picture
Update app.py
bf957dc
raw
history blame contribute delete
306 Bytes
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)