Update app.py

app.py

@@ -96,17 +96,14 @@ if st.button("Predict"):
         Oldpeak=oldpeak,
         ST_Slope=st_slope
     )
-    
-    # Send a request to the FastAPI server
-    response = predict(data)
-
-    # Display the result
-    if response.status_code == 200:
-        prediction = response.json()["prediction"]
-        result = "Positive for heart disease" if prediction == 1 else "Negative for heart disease"
-        st.success(f"Prediction: {result}")
+
+    result = predict(data)
+
+    st.write("### Prediction:")
+    if result == 1:
+        st.write("The model predicts a high risk of heart disease.")
     else:
-        st.error("Error: Unable to get prediction from API. Please try again later.")
+        st.write("The model predicts a low risk of heart disease.")
 
 
 st.subheader("Batch Prediction")