Update main.py

main.py

@@ -27,26 +27,24 @@ app.add_middleware(
 
 # Endpoint for making predictions
 @app.post("/predict")
-def predict(customer_name: str,
+def predict(
+    customer_name: str,
     customer_address: str,
     customer_phone: str,
     customer_email: str,
-    cod:str,
     weight: str,
     pickup_address: str,
-    origin_city_name: str,
-    destination_city_name: str,
-    origin_country: str):
+    destination_city_name: str):
     # Convert input data to DataFrame
     input_data = {
         'customer_name': customer_name,
         'customer_address': customer_address,
         'customer_phone': customer_phone,
         'customer_email': customer_email,
-        'cod': float(cod),
+        'cod': 1.0,
         'weight': float(weight),
         'pickup_address':pickup_address,
-        'origin_city.name':origin_city_name,
+        'origin_city.name':"origin_city_name",
         'destination_city.name':destination_city_name
     }
     input_df = pd.DataFrame([input_data])