Change output

app.py

@@ -41,7 +41,8 @@ def predict(context, intent):
     pn_tensor = F.normalize(pn_tensor, p=1, dim=0)
 
     pn_tensor = pn_tensor.softmax(dim=0)
-    return {"entailment": pn_tensor[0].item(), "neutral": pn_tensor[1].item(), "contradiction": pn_tensor[2].item()}
+    pn_tensor = pn_tensor.tolist()
+    return {"entailment": pn_tensor[0], "neutral": pn_tensor[1], "contradiction": pn_tensor[2]}
 
 gradio_app = gr.Interface(
     predict,