Update data_preparation.py

data_preparation.py

@@ -98,12 +98,8 @@ def data_imp():
 def preprocess_data(data):
     if 'CustID' in data.columns:
         data = data.drop(columns=['CustID'])
-    if 'Channel' in data.columns:
-        data = data.drop(columns=['Channel'])
-    if 'Region' in data.columns:
-        data = data.drop(columns=['Region'])
-    
-
+    if 'Cust_ID' in data.columns:
+        data = data.drop(columns=['Cust_ID'])
     data = remove_outliers(data)
     return data