changed outputs

app.py

@@ -23,8 +23,12 @@ def classify_text(text):
     # Convert the index to label
     label = "ham" if prediction_index == 0 else "spam"
 
-    # Return the label and the probabilities of each class
-    return {"label": label, "probabilities": probabilities.numpy().tolist()}
+    # Get the probabilities of each class
+    ham_prob = probabilities[0][0].numpy()
+    spam_prob = probabilities[0][1].numpy()
+
+    # Return the label and the probabilities separately
+    return label, {"ham": ham_prob, "spam": spam_prob}
 
 # Create the Gradio interface
 iface = gr.Interface(