Update app.py
Browse files
app.py
CHANGED
@@ -246,6 +246,7 @@ def process_dataframe(df):
|
|
246 |
|
247 |
# Create two DataFrames: one for prediction and one for classification.
|
248 |
df_pred = df[required_columns].copy()
|
|
|
249 |
df_class = df[required_columns_2].fillna("NA").copy()
|
250 |
|
251 |
# Transform categorical columns for prediction DataFrame using the label encoders.
|
|
|
246 |
|
247 |
# Create two DataFrames: one for prediction and one for classification.
|
248 |
df_pred = df[required_columns].copy()
|
249 |
+
df_pred[['EngBlk', 'EngWht', 'EngOpen', 'EngPav']]=df_pred[['EngBlk', 'EngWht', 'EngOpen', 'EngPav']].fillna("NA")
|
250 |
df_class = df[required_columns_2].fillna("NA").copy()
|
251 |
|
252 |
# Transform categorical columns for prediction DataFrame using the label encoders.
|