diff --git a/streamlit/interface_3.py b/streamlit/interface_3.py index 63184b1..e0f3f8a 100644 --- a/streamlit/interface_3.py +++ b/streamlit/interface_3.py @@ -83,7 +83,7 @@ try: fields.rename(columns={'QUESTION': 'Interrogation Question?'}, inplace = True) fields.rename(columns={'PRIORITY': 'PRIORITY'}, inplace = True) fields.rename(columns={'FIELD_NAME': 'SF_DB_COL_NAME'}, inplace = True) - fields = fields[~fields[column_name].str.endswith('_PG', na=None)] + fields = fields[~fields['SF_DB_COL_NAME'].str.endswith('_PG', na=None)] fields['Field Name'] = fields['SF_DB_COL_NAME'] # fields.rename(columns={'FM_MODEL_ID': 'llm_selected'}, inplace = True) # error('table is empty')