Update app.py
Browse files
app.py
CHANGED
@@ -115,24 +115,26 @@ if st.button("Predict"):
|
|
115 |
}
|
116 |
|
117 |
# Send a POST request to the FastAPI server
|
118 |
-
|
119 |
|
120 |
-
|
121 |
-
|
|
|
122 |
prediction_data = response.json()
|
123 |
-
|
124 |
# Display prediction result to the user
|
125 |
st.subheader("Prediction Result")
|
126 |
|
127 |
# Determine income prediction and format message
|
128 |
if prediction_data['income_prediction'] == "Income over $50K":
|
129 |
-
|
130 |
else:
|
131 |
-
|
132 |
-
|
133 |
# Display prediction probability
|
134 |
st.subheader("Prediction Probability")
|
135 |
probability = prediction_data['prediction_probability']
|
136 |
st.write(f"The probability of the individual having an income over $50K is: {probability:.2f}")
|
137 |
-
|
138 |
-
st.error("Error: Unable to get prediction")
|
|
|
|
115 |
}
|
116 |
|
117 |
# Send a POST request to the FastAPI server
|
118 |
+
response = requests.post("https://rasmodev-income-prediction-fastapi.hf.space/predict/", json=user_input)
|
119 |
|
120 |
+
|
121 |
+
# Check if the request was successful
|
122 |
+
if response.status_code == 200:
|
123 |
prediction_data = response.json()
|
124 |
+
|
125 |
# Display prediction result to the user
|
126 |
st.subheader("Prediction Result")
|
127 |
|
128 |
# Determine income prediction and format message
|
129 |
if prediction_data['income_prediction'] == "Income over $50K":
|
130 |
+
st.success("This individual is predicted to have an income of over $50K.")
|
131 |
else:
|
132 |
+
st.error("This individual is predicted to have an income of under $50K")
|
133 |
+
|
134 |
# Display prediction probability
|
135 |
st.subheader("Prediction Probability")
|
136 |
probability = prediction_data['prediction_probability']
|
137 |
st.write(f"The probability of the individual having an income over $50K is: {probability:.2f}")
|
138 |
+
else:
|
139 |
+
st.error("Error: Unable to get prediction")
|
140 |
+
|