Update app.py

fix pred feature

app.py

@@ -108,8 +108,8 @@ def predict(features: PredictionFeatures):
     
     try:
         df = pd.DataFrame({...})
-        prediction = model.predict(df)[0]
-        return {"prediction": int(prediction[0])}
+        return model.predict(df)[0]
+    
     except Exception as e:
         return {"error": str(e)}