gdleds commited on
Commit
a046c28
·
1 Parent(s): b7cb626

troisieme

Browse files
Files changed (1) hide show
  1. app.py +36 -12
app.py CHANGED
@@ -39,22 +39,46 @@ model = load_model_from_s3(S3_BUCKET, MODEL_KEY)
39
  class InputData(BaseModel):
40
  input: list
41
 
42
-
43
  @app.post("/predict")
44
  def predict(data: InputData):
45
- # Colonnes attendues par le modèle
46
- columns = [
47
- "mileage", "engine_power", "fuel", "paint_color", "car_type",
48
- "private_parking_available", "has_gps", "has_air_conditioning",
49
- "automatic_car", "has_getaround_connect", "has_speed_regulator", "winter_tires"
50
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
 
52
- # Transformer l'input en DataFrame
53
- X = pd.DataFrame(data.input, columns=columns)
54
 
55
- # Prédictions
56
- preds = model.predict(X)
57
- return {"prediction": preds.tolist()}
58
 
59
 
60
  # @app.post("/predict")
 
39
  class InputData(BaseModel):
40
  input: list
41
 
 
42
  @app.post("/predict")
43
  def predict(data: InputData):
44
+ try:
45
+ columns = [
46
+ "mileage", "engine_power", "fuel", "paint_color", "car_type",
47
+ "private_parking_available", "has_gps", "has_air_conditioning",
48
+ "automatic_car", "has_getaround_connect", "has_speed_regulator", "winter_tires"
49
+ ]
50
+
51
+ print("📥 Input reçu :", data.input) # log input brut
52
+ X = pd.DataFrame(data.input, columns=columns)
53
+ print("✅ DataFrame construit :", X.head().to_dict()) # log input formaté
54
+
55
+ preds = model.predict(X)
56
+ print("📤 Prediction faite :", preds)
57
+
58
+ return {"prediction": preds.tolist()}
59
+
60
+ except Exception as e:
61
+ import traceback
62
+ print("❌ Erreur lors de la prédiction :", e)
63
+ print(traceback.format_exc())
64
+ return {"error": str(e)}
65
+
66
+
67
+ # @app.post("/predict")
68
+ # def predict(data: InputData):
69
+ # # Colonnes attendues par le modèle
70
+ # columns = [
71
+ # "mileage", "engine_power", "fuel", "paint_color", "car_type",
72
+ # "private_parking_available", "has_gps", "has_air_conditioning",
73
+ # "automatic_car", "has_getaround_connect", "has_speed_regulator", "winter_tires"
74
+ # ]
75
 
76
+ # # Transformer l'input en DataFrame
77
+ # X = pd.DataFrame(data.input, columns=columns)
78
 
79
+ # # Prédictions
80
+ # preds = model.predict(X)
81
+ # return {"prediction": preds.tolist()}
82
 
83
 
84
  # @app.post("/predict")