Update app.py

app.py

@@ -6,9 +6,11 @@ token = os.environ["token"]
 classifier = pipeline("zero-shot-classification",
                       model="joeddav/xlm-roberta-large-xnli",use_auth_token=token)
 
-def classify(text):
-    return classifier(text, candidate_labels)
+def classify(text, intent_labels):
+    res = classifier(text, intent_labels)
+    results = {k: v for k, v in zip(res["labels"], res["scores"])}
+    return results
 
-candidate_labels = ["wasser", "strom", "gas", "hausreinigung", "schoenheit", "wellness", "oeffentlichkeit", "oeffentlicher", "verkehr", "vermietung", "medizin", "gesundheit", "pflege", "reisen", "tourismus", "versicherungen", "recht"]
+# input is two text boxes, one for the text and one for the labels
 
-gr.Interface(fn=classify, inputs="text", outputs="label").launch()
+gr.Interface(fn=classify, inputs=["textbox", "textbox"], outputs="label").launch()