Update app.py

app.py

@@ -44,6 +44,7 @@ async def predict(data: InputData):
             1, -1
         )  # Asumiendo que la entrada debe ser de forma (1, num_features)
         prediction = model.predict(input_data).round()
-        return {"prediction": prediction.tolist()}
+        #return {"prediction": prediction.tolist()}
+        return {"prediction": prediction[0][0]}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))