albin commited on
Commit
15559a4
·
1 Parent(s): 9372d92
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -123,6 +123,8 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
123
 
124
  # Check number of dimensions before prediction
125
  print("Nb dimensions before prediction:", data.shape)
 
 
126
 
127
  prediction = model.predict([data])
128
  output = prediction[0]
 
123
 
124
  # Check number of dimensions before prediction
125
  print("Nb dimensions before prediction:", data.shape)
126
+ print("Data sent to predict:", data)
127
+ print("Data types:", dataFrame_features[selected_columns].dtypes)
128
 
129
  prediction = model.predict([data])
130
  output = prediction[0]