Update app.py

app.py

@@ -117,23 +117,5 @@ if st.button("Predict"):
     }
     # Send a POST request to the FastAPI server
     response = requests.post("https://rasmodev-income-prediction-fastapi.hf.space/predict", json=user_input)
-    
-    # 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")
+    prediction = response.text
+    st.success(f'This individual is predicted to have an income of:{prediction})