pradelf commited on
Commit
18c9ca8
·
verified ·
1 Parent(s): 46bb294

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -153,7 +153,7 @@ async def predict(predictionFeatures: RentalFeatures):
153
  # ]
154
  try:
155
  payload = predictionFeatures.model_dump()
156
- car_caracteristic = pd.DataFrame([payload])
157
  #car_caracteristic = pd.DataFrame({"Car": pf})
158
  # Log model from mlflow
159
  BOOL_COLS = [
@@ -167,14 +167,14 @@ async def predict(predictionFeatures: RentalFeatures):
167
  ]
168
  # Conversion explicite des colonnes booléennes
169
  for col in BOOL_COLS:
170
- car_caracteristic[col] = car_caracteristic[col].astype(bool)
171
 
172
 
173
  logger.info("Payload reçu: %s", payload)
174
  logger.info("dtypes:\n%s", car_characteristic.dtypes)
175
 
176
  # Prediction from previously loaded model as a PyFuncModel.
177
- prediction = model.predict(car_caracteristic)
178
  logger.info(f"Output : {prediction}")
179
  # Format response
180
  response = {
 
153
  # ]
154
  try:
155
  payload = predictionFeatures.model_dump()
156
+ car_characteristic = pd.DataFrame([payload])
157
  #car_caracteristic = pd.DataFrame({"Car": pf})
158
  # Log model from mlflow
159
  BOOL_COLS = [
 
167
  ]
168
  # Conversion explicite des colonnes booléennes
169
  for col in BOOL_COLS:
170
+ car_characteristic[col] = car_characteristic[col].astype(bool)
171
 
172
 
173
  logger.info("Payload reçu: %s", payload)
174
  logger.info("dtypes:\n%s", car_characteristic.dtypes)
175
 
176
  # Prediction from previously loaded model as a PyFuncModel.
177
+ prediction = model.predict(car_characteristic)
178
  logger.info(f"Output : {prediction}")
179
  # Format response
180
  response = {