Trying zero-shot classification.

app.py

@@ -1,10 +1,15 @@
 import gradio
-def zero_shot_classification(name):
-    return "Hello " + name + "!!"
+from transformers import pipeline
+
+def zero_shot_classification(data):
+    classifier = pipeline('zero-shot-classification, model='Xenova/mobilebert-uncased-mnli')
+
+    results = classifier(data.sequence, candidate_labels=data.candidate_labels, hypothesis_template=data.hypothesis_template, multi_label=data.multi_label)
+    return {results}
 
 gradio_interface = gradio.Interface(
     fn = zero_shot_classification,
-    inputs = "text",
-    outputs = "text"
+    inputs = "json",
+    outputs = "json"
 )
 gradio_interface.launch()