Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -45,6 +45,7 @@ async def predict(data: InputData):
|
|
45 |
) # Asumiendo que la entrada debe ser de forma (1, num_features)
|
46 |
prediction = model.predict(input_data).round()
|
47 |
#return {"prediction": prediction.tolist()}
|
|
|
48 |
return {"prediction": prediction[0][0]}
|
49 |
except Exception as e:
|
50 |
raise HTTPException(status_code=500, detail=str(e))
|
|
|
45 |
) # Asumiendo que la entrada debe ser de forma (1, num_features)
|
46 |
prediction = model.predict(input_data).round()
|
47 |
#return {"prediction": prediction.tolist()}
|
48 |
+
print(prediction)
|
49 |
return {"prediction": prediction[0][0]}
|
50 |
except Exception as e:
|
51 |
raise HTTPException(status_code=500, detail=str(e))
|