WebashalarForML commited on
Commit
54be9fd
·
verified ·
1 Parent(s): fa58961

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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.