Update app.py

app.py

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