Update app.py

app.py

@@ -7,15 +7,19 @@ classifier = pipeline("zero-shot-classification",
                       model=model, tokenizer=tokenizer)
 
 def classify(input_query, input_classes, input_multi_class):
-  result = []
-  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')))}
+    
+    input_candidate_classes = input_classes
+
+    res = classifier(input_query, input_candidate_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'), gr.Checkbox(label='Multi-class')],
-  outputs=gr.Label(label='prediction:')
+  inputs=[gr.Text(label='Search Query'), gr.Text(label='Candidate Classes'), gr.Checkbox(label='Multi_class')],
+  outputs=gr.Label(label='Prediction:')
 )
 
 demo.launch()