New model with json output

app.py

@@ -5,7 +5,6 @@ os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 from transformers import AutoTokenizer, TFAutoModelForSequenceClassification, TextClassificationPipeline
 
-
 tokenizer = AutoTokenizer.from_pretrained("dipesh/Intent-Classification-Bert-Base-Cased")
 
 model = TFAutoModelForSequenceClassification.from_pretrained("dipesh/Intent-Classification-Bert-Base-Cased")
@@ -25,9 +24,10 @@ def predict(input_text):
         ans += "?"
     ans = intent_classifier(input_text)
 
-    return ans[0]['label'], ans[0]['score']
+    return {"class": ans[0]['label'],
+            "accuracy": ans[0]['score']}
 
 
-iface = gr.Interface(fn=predict, inputs="text", outputs="text", title="Intent Classifier",
+iface = gr.Interface(fn=predict, inputs="text", outputs="json", title="Intent Classifier",
                      description="Classifier")
 iface.launch()