Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -153,7 +153,7 @@ async def predict(predictionFeatures: RentalFeatures):
|
|
| 153 |
# ]
|
| 154 |
try:
|
| 155 |
payload = predictionFeatures.model_dump()
|
| 156 |
-
|
| 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 |
-
|
| 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(
|
| 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 = {
|