Update main.py

main.py

@@ -35,7 +35,9 @@ def predict(customer_name: str,
     weight: str,
     pickup_address: str,
     origin_city_name: str,
-    destination_city_name: str):
+    destination_city_name: str,
+    destination_country: str,
+    rigin_country: str):
     # Convert input data to DataFrame
     input_data = {
         'customer_name': customer_name,