Spaces:

Alex3456
/

ZeroShotClassification

Runtime error

Update app.py

a01ae42 almost 3 years ago

324 Bytes

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)

	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)