Update main.py

main.py

@@ -60,6 +60,10 @@ async def predict_income(data: IncomePredictionRequest):
     prediction_result = "Income over $50K" if prediction[0] == 1 else "Income under $50K"
     return {"income_prediction": prediction_result, "prediction_probability": prediction_proba[0][1]}
 
+    except Exception as e:
+        logging.error(f"Prediction failed: {e}")
+        raise
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)