Playing with different models.

app.py

@@ -10,11 +10,11 @@ def zero_shot_classification(data_string):
     data = json.loads(data_string)
     print(data)
     results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
-    return results
+    return json.dumps(results)
 
 gradio_interface = gradio.Interface(
     fn = zero_shot_classification,
     inputs = gradio.Textbox(label="JSON Input"),
-    outputs = "text"
+    outputs = "json"
 )
 gradio_interface.launch()