Update app.py

app.py

@@ -41,7 +41,7 @@ def predict_email(text):
         prob_spam = model(X_tensor).item()
         
     # Manually calculate Ham probability as the complement of Spam
-    return {"Ham": 1 - prob_spam, "Spam": prob_spam}
+    return {"Safe": 1 - prob_spam, "Spam": prob_spam}
 
 # ---------- Gradio Interface ----------
 iface = gr.Interface(