Update pages/1_Clothing_Bias.py

pages/1_Clothing_Bias.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import string as s
 from PIL import Image
 import requests
 from io import BytesIO
@@ -16,7 +17,7 @@ def load_clip_model():
     return model, processor
 
 # --- Main page content ---
-st.header("👖 Clothing Bias in Scene Classification 👗")
+st.header("Clothing Bias in Scene Classification")
 st.markdown("""
 This application explores biases in scene classification models related to clothing attributes.
 It leverages the CLIP model to analyze and highlight these biases.
@@ -50,8 +51,8 @@ elif input_method == "Default Images":
         col = columns[i % 4]
         img_path = os.path.join(image_dir, image_file)
         with col:
-            st.image(img_path, caption=image_file, use_column_width=True)
-            if st.button(f"Select {image_file}", key=image_file):
+            st.image(img_path, caption=image_file, use_container_width=True)
+            if st.button(f"Select {s.ascii_uppercase[i]}", key=image_file):
                 selected_image = image_file
 
     # Store selected image using session state so selection persists
@@ -69,7 +70,7 @@ if image is not None:
 
 # Prompt input
 st.subheader("📝 Candidate Scene Labels")
-default_prompts = ["a business executive", "a festival participant"]
+default_prompts = ["a business executive", "a festival participant", "a indian maid", "a school teacher"]
 prompts_text = st.text_area("Enter one label per line:", "\n".join(default_prompts))
 labels = [label.strip() for label in prompts_text.strip().split("\n") if label.strip()]