Update app.py

app.py

@@ -304,10 +304,15 @@ def process_dataframe(df):
             df_pred_0['Change_Blk_Eng_to_Mkbl_value'] = pd.DataFrame(blk_change.predict(df_pred), columns=["Change_Blk_Eng_to_Mkbl_value"])
             print(df_pred_0.columns)
 
+            # for model SHP CODE (need change)
+            df_pred_0['Change_shape_value'] = pd.DataFrame(shape_change.predict(df_pred), columns=["Change_shape_value"])
+            print(df_pred_0.columns)
+            
+            '''
             # for model WHT CODE
             df_pred_0['Change_Wht_Eng_to_Mkbl_value'] = pd.DataFrame(shape_change.predict(df_pred), columns=["Change_Wht_Eng_to_Mkbl_value"])
             print(df_pred_0.columns)
-            '''
+
             # for model PAV CODE (need change)
             df_pred_0['Change_Pav_Eng_to_Mkbl_value'] = pd.DataFrame(mkble_amt_class_model.predict(df_pred), columns=["Change_Pav_Eng_to_Mkbl_value"])
             print(df_pred_0.columns)
@@ -316,10 +321,6 @@ def process_dataframe(df):
             df_pred_0['Change_Open_Eng_to_Mkbl_value'] = pd.DataFrame(mkble_amt_class_model.predict(df_pred), columns=["Change_Open_Eng_to_Mkbl_value"])
             print(df_pred_0.columns)
 
-            # for model SHP CODE (need change)
-            df_pred_0['Change_shape_value'] = pd.DataFrame(mkble_amt_class_model.predict(df_pred), columns=["Change_shape_value"])
-            print(df_pred_0.columns)
-
             # for model COL CODE (need change)
             df_pred_0['Change_color_value'] = pd.DataFrame(mkble_amt_class_model.predict(df_pred), columns=["Change_color_value"])
             print(df_pred_0.columns)
@@ -338,13 +339,17 @@ def process_dataframe(df):
     
             for col in ['Tag', 'EngShp', 'EngQua', 'EngCol', 'EngCut', 'EngPol', 'EngSym', 'EngFlo',
                         'EngNts', 'EngMikly','EngBlk', 'EngWht', 'EngOpen','EngPav',
-                        'Change_cts_value', 'Change_shape_value', 'Change_quality_value', 'Change_color_value',
-                        'Change_cut_value', 'Change_Blk_Eng_to_Mkbl_value', 'Change_Wht_Eng_to_Mkbl_value',
-                        'Change_Open_Eng_to_Mkbl_value', 'Change_Pav_Eng_to_Mkbl_value', 'Change_Blk_Eng_to_Grd_value',
-                        'Change_Wht_Eng_to_Grd_value', 'Change_Open_Eng_to_Grd_value', 'Change_Pav_Eng_to_Grd_value',
-                        'Change_Blk_Eng_to_ByGrd_value', 'Change_Wht_Eng_to_ByGrd_value', 'Change_Open_Eng_to_ByGrd_value',
-                        'Change_Pav_Eng_to_ByGrd_value', 'Change_Blk_Eng_to_Gia_value', 'Change_Wht_Eng_to_Gia_value',
-                        'Change_Open_Eng_to_Gia_value', 'Change_Pav_Eng_to_Gia_value']:
+                        'Change_shape_value',
+                        #'Change_cts_value','Change_quality_value', 'Change_color_value', 'Change_cut_value',
+                        'Change_Blk_Eng_to_Mkbl_value', 
+                        #'Change_Wht_Eng_to_Mkbl_value',
+                        #'Change_Open_Eng_to_Mkbl_value',
+                        #'Change_Pav_Eng_to_Mkbl_value', 
+                        #'Change_Blk_Eng_to_Grd_value','Change_Wht_Eng_to_Grd_value', 'Change_Open_Eng_to_Grd_value', 'Change_Pav_Eng_to_Grd_value',
+                        #'Change_Blk_Eng_to_ByGrd_value', 'Change_Wht_Eng_to_ByGrd_value', 'Change_Open_Eng_to_ByGrd_value', 'Change_Pav_Eng_to_ByGrd_value', 
+                        #'Change_Blk_Eng_to_Gia_value', 'Change_Wht_Eng_to_Gia_value', 'Change_Open_Eng_to_Gia_value', 'Change_Pav_Eng_to_Gia_value'
+                       ]:
+                
                 try:
                     df_pred_main[col] = loaded_label_encoder[col].inverse_transform(df_pred_main[col].astype(int))
                 except ValueError as e: