Update app.py

app.py

@@ -43,7 +43,7 @@ async def predict(data: InputData):
         prediction = miarbol.predict(X_input)
 
         # Retornar el resultado de la predicción
-        return {"prediction": (int)prediction[0]}
+        return {"prediction": prediction[0]}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))