Update app.py

app.py

@@ -65,6 +65,7 @@ app = FastAPI(
 
 
 class RentalFeatures(BaseModel):
+    model_key: str
     mileage: int
     engine_power: int
     fuel: str
@@ -102,6 +103,7 @@ async def predict(predictionFeatures: RentalFeatures):
     """
     # Read data
     pf = [
+        predictionFeatures.model_key or "Peugeot",
         predictionFeatures.mileage or 0,
         predictionFeatures.engine_power or 0,
         predictionFeatures.fuel or "petrol",