Update main.py

main.py

@@ -54,9 +54,9 @@ async def train_the_model(Tenant: str):
         data = oversampled_data
         
 
-        # Select columns
+        # Select columns 'customer_email'
         selected_columns = ['customer_name', 'customer_address', 'customer_phone',
-                            'customer_email', 'cod', 'weight', 'origin_city.name',
+                            'cod', 'weight', 'origin_city.name',
                             'destination_city.name','created_at','status.name']
         
         # Handling missing values
@@ -223,8 +223,7 @@ def predict(
     input_data = {
         'customer_name': customer_name,
         'customer_address': customer_address,
-        'customer_phone': int(customer_phone),
-        'customer_email': customer_email,
+        'customer_phone': int(customer_phone), #'customer_email': customer_email,
         'cod': float(cod),
         'weight': float(weight),
         'origin_city.name':origin_city_name,