Create app.py

app.py

@@ -0,0 +1,17 @@
+from transformers import pipeline
+classifier = pipeline("zero-shot-classification",
+                      model="facebook/bart-large-mnli")
+
+def classify(input_query, input_classes):
+  result = []
+  res = classifier(sequence_to_classify, input_classes)
+  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')],
+  outputs=gr.Label(label='prediction:')
+)
+
+demo.launch()