Update app.py

app.py

@@ -28,8 +28,10 @@ def cleaning_features(data):
     # with open('race_onehot_encoder.pkl', 'rb') as enc_file:
     #     encoder = pickle.load(enc_file)
     
-    with open('label_encoder.pkl', 'rb') as le_file:
-        le = pickle.load(le_file)
+    with open('label_encoder_work.pkl', 'rb') as le_file:
+        le_work = pickle.load(le_file)
+    with open('label_encoder_occ.pkl', 'rb') as le_file:
+        le_occ = pickle.load(le_file)
 
     with open('scaler.pkl', 'rb') as scaler_file:
         scaler = pickle.load(scaler_file)
@@ -59,8 +61,8 @@ def cleaning_features(data):
     numeric_cols = ['age', 'educational-num', 'hours-per-week']
     columns_to_encode = ['race','marital-status','relationship']
     
-    data['workclass'] = le.transform(data['workclass'])
-    data['occupation'] = le.transform(data['occupation'])
+    data['workclass'] = le_work.transform(data['workclass'])
+    data['occupation'] = le_occ.transform(data['occupation'])
     data['gender'] = data['gender'].map(gender_mapping)
     data['native-country'] = data['native-country'].map(country_mapping)
     data['educational-num'] = data['educational-num'].map(education_num_mapping)