RaulHuarote commited on
Commit
1ac9e6f
·
verified ·
1 Parent(s): b8c76b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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))