Trying zero-shot classification.

app.py

@@ -1,7 +1,8 @@
 import gradio
 from transformers import pipeline
 
-def zero_shot_classification(data):
+def zero_shot_classification(data_string):
+    data = json.loads(data_string)
     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)
@@ -9,7 +10,7 @@ def zero_shot_classification(data):
 
 gradio_interface = gradio.Interface(
     fn = zero_shot_classification,
-    inputs = "json",
+    inputs = "text",
     outputs = "json"
 )
 gradio_interface.launch()