Update app.py

app.py

@@ -53,6 +53,16 @@ def predict_email(email):
     prediction = obj.predict(data)[0]
     st.write(f"Prediction: {prediction}")
 
+@app.get("/predict/")
+def predict_spam(email: str):
+    """
+    This endpoint predicts whether the email is Spam or Ham.
+    Query parameter: email (str) - The email text to be classified.
+    """
+    data = bow.transform([email]).toarray()
+    prediction = obj.predict(data)[0]
+    return JSONResponse(content={"prediction": prediction})    
+
 
 if st.button("Predict Email"):
     if email_input: