Update app/main.py

app/main.py

@@ -24,6 +24,8 @@ def predict(df, endpoint="simple"):
 
     response = []
     for eta in prediction:
+        # Convert NumPy float to Python native float
+        eta = float(eta)
         # Create a response for each prediction with the predicted ETA
         output = {
             "predicted_eta": eta
@@ -74,7 +76,7 @@ def predict_eta(trip: Trip):
     # Make prediction
     data = pd.DataFrame(trip.dict(), index=[0])
     predicted_eta = predict(df=data)
-    return {"predicted_eta": predicted_eta}
+    return predicted_eta
 
 # Multiple Prediction Endpoint
 @app.post("/predict_multiple")