Trying zero-shot classification.

app.py

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