Update app.py

app.py

@@ -102,8 +102,12 @@ def predict(features: PredictionFeatures):
         "age_fraud_level" : [features.age_fraud_level]
     })
     
-    prediction = model.predict(df)[0]
-    return {"prediction": float(prediction)}
+    try:
+        df = pd.DataFrame({...})
+        prediction = model.predict(df)[0]
+        return {"prediction": int(prediction[0])}
+    except Exception as e:
+        return {"error": str(e)}
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)