Update main.py

main.py

@@ -32,8 +32,8 @@ def predict(
     customer_address: str,
     customer_phone: str,
     customer_email: str,
-    weight: str,
-    cod:str,
+    weight: int,
+    cod: int,
     pickup_address: str,
     destination_city_name: str):
     # Convert input data to DataFrame
@@ -46,8 +46,8 @@ def predict(
         'customer_address': customer_address,
         'customer_phone_no': customer_phone,
         'client_email': customer_email,
-        'weight': float(weight),
-        'cod': float(cod),
+        'weight': weight,
+        'cod': cod,
         'pickup_address':pickup_address,
         'destination_branch_name':destination_city_name
     }
@@ -65,6 +65,7 @@ def predict(
     # Output
     predicted_status = "Unknown" if pred[0] == -1 else encoders['status_name'].inverse_transform([pred])[0]
     probability = pred_proba[0][pred[0]] * 100 if pred[0] != -1 else "Unknown"
+    print(input_data,predicted_status,probability)
 
     if predicted_status == "Returned to Client":
        probability = 100 - probability