modifying the prediction view

app.py

@@ -463,12 +463,12 @@ if st.button('Generate Predictions'):
                     sel_var_values.append(round(selected_variable_number, 3)*100)
                     # st.write(f"·Number of Characters: {int(Character_Count)}, Target Rate: {round(round(selected_variable_number, 3)*100, 3)}", "%")
                 st.write("\n")
-                df_modelpred=pd.DataFrame(list(zip(chars, sel_var_values)), columns=["Number of Characters", "Target Rate"])
+                df_modelpred=pd.DataFrame(list(zip(chars, sel_var_values)), columns=["Number of Characters", "Target_Rate"])
                 # st.checkbox("Use container width", value=False, key="use_container_width")
                 # st.dataframe(df_modelpred.style.highlight_max(axis=0), use_container_width=st.session_state.use_container_width)
                 df_modelpred.sort_values(by='Target Rate',inplace = True)
                 def color_coding(row):
-                    return ['background-color:yellow'] * len(row) if row.col2 ==df_modelpred['Target Rate'][0]  else ['background-color:yellow'] * len(row)
+                    return ['background-color:yellow'] * len(row) if row.Target_Rate ==df_modelpred['Target_Rate'][0]  else ['background-color:yellow'] * len(row)
                 st.dataframe(df_modelpred.style.apply(color_coding, axis=1))
 
                 if len(chars) > 1: