Update app.py

Add multi_class

app.py

@@ -2,15 +2,15 @@ from transformers import pipeline
 classifier = pipeline("zero-shot-classification",
                       model="facebook/bart-large-mnli")
 
-def classify(input_query, input_classes):
+def classify(input_query, input_classes, input_multi_class):
   result = []
-  res = classifier(sequence_to_classify, input_classes)
+  res = classifier(sequence_to_classify, input_classes, multi_class=input_multi_class)
   res_dict = {res.get('labels')[i]: res.get('scores')[i] for i in range(len(res.get('labels')))}
   return res_dict
 
 demo = gr.Interface(
   fn=classify,
-  inputs=[gr.Text(label='Search Query'), gr.Text(label='Candidate Classes')],
+  inputs=[gr.Text(label='Search Query'), gr.Text(label='Candidate Classes'), gr.Checkbox(label='Multi-class')],
   outputs=gr.Label(label='prediction:')
 )