Update app.py

app.py

@@ -1017,7 +1017,6 @@ if analysis_option == 'Machine Learning':
     token = os.environ["HUGGING_FACE_HUB_TOKEN"]
     local_file_1 = hf_hub_download(repo_id=repo_id, filename=filename_1,repo_type="dataset", token=token)
     filtered_data = pd.read_parquet(local_file_1)
-    st.write(filtered_data.columns)
     if filtered_data['key_lab2'].notna().any():
         column_list = ['PatientID', 'VisitID', 'GroupedICD'] + list(filtered_data['key_lab2'].iloc[0])
         pivot_data = pd.pivot_table(filtered_data, values='ComponentValue', index=['PatientID', 'VisitID', 'GroupedICD'], columns='ComponentName', aggfunc=lambda x: ', '.join(map(str, x)))