Update app.py

app.py

@@ -212,7 +212,9 @@ with tab3:
     edited_df = st.data_editor(ef_df, num_rows="dynamic")
 
     if st.button("Convert to Binary Labels"):
-        common_cols = list(set(edited_df.columns) & set(thresholds.index))
+        common_cols = list(set(map(int, edited_df.columns)) & set(thresholds.index))
+        common_cols = list(map(str, common_cols))  # Convert back to string for use as column names
+
         numeric_cols = edited_df[common_cols].select_dtypes(include=[np.number]).columns.tolist()
     
         binary_part = edited_df[numeric_cols].ge(thresholds[numeric_cols]).astype(int)