Update app.py

app.py

@@ -118,25 +118,24 @@ 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)
+    response = requests.post("http://localhost:8000/predict/", json=user_input)  # Use the correct URL
 
-    
     # 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}")
+    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("Error: Unable to get prediction")
+            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")