Update app.py

app.py

@@ -59,14 +59,14 @@ def cleaning_features(data):
     numeric_cols = ['age', 'educational-num', 'hours-per-week']
     columns_to_encode = ['race','marital-status','relationship']
     
-    data['gender'] = le.fit_transform(data['gender'])
-    data['workclass'] = le.fit_transform(data['workclass'])
-    data['occupation'] = le.fit_transform(data['occupation'])
+    data['gender'] = le.transform(data['gender'])
+    data['workclass'] = le.transform(data['workclass'])
+    data['occupation'] = le.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)
     
-    data[numeric_cols] = scaler.fit_transform(data[numeric_cols])
+    data[numeric_cols] = scaler.transform(data[numeric_cols])
 
     #data = pca(data)
     return data