Update app.py

app.py

@@ -46,7 +46,7 @@ def call_pricing_api(featRentalFeatures: RentalFeatures):
         data = resp.json()
         logger.info(data)
         if "prediction" in data:
-            return float(data["prediction"][0])
+            return float(data["prediction"])
         else:
             st.error("Réponse API inattendue : clé 'prediction' absente.")
             return None, payload, data