Update app.py

app.py

@@ -115,7 +115,28 @@ if st.button("Predict"):
         "country_of_birth_own": country_of_birth_own,
         "importance_of_record": float(importance_of_record)
     }
+
+    
     # Send a POST request to the FastAPI server
     response = requests.post("https://rasmodev-income-prediction-fastapi.hf.space/predict", json=user_input)
-    prediction = response.text
-    st.success(f"This individual is predicted to have an income of:{prediction}")
+
+    
+    # Check if the request was successful
+        if response.status_code == 200:
+            prediction_data = response.json()
+            
+            # Display prediction result to the user
+            st.subheader("Prediction Result")
+        
+            # Determine income prediction and format message
+            if prediction_data['income_prediction'] == "Income over $50K":
+                st.success("This individual is predicted to have an income of over $50K.")
+            else:
+                st.error("This individual is predicted to have an income of under $50K")
+            
+            # Display prediction probability
+            st.subheader("Prediction Probability")
+            probability = prediction_data['prediction_probability']
+            st.write(f"The probability of the individual having an income over $50K is: {probability:.2f}")
+        else:
+            st.error("Error: Unable to get prediction")