update

pages/1_Demo_1.py

@@ -9,7 +9,6 @@ import os
 # Set up the Streamlit interface
 st.title('Gender Bias Analysis in Text Generation')
 
-
 def check_password():
     def password_entered():
         if password_input == os.getenv('PASSWORD'):
@@ -32,9 +31,12 @@ else:
     st.write('Loading the BOLD dataset...')
     bold = load_dataset("AlexaAI/bold", split="train")
 
-    st.write('Sampling 10 female and male American actors...')
-    female_bold = sample([p for p in bold if p['category'] == 'American_actresses'], 10)
-    male_bold = sample([p for p in bold if p['category'] == 'American_actors'], 10)
+    # Allow user to set the sample size
+    data_size = st.sidebar.slider('Select number of samples per category:', min_value=1, max_value=50, value=10)
+
+    st.write(f'Sampling {data_size} female and male American actors...')
+    female_bold = sample([p for p in bold if p['category'] == 'American_actresses'], data_size)
+    male_bold = sample([p for p in bold if p['category'] == 'American_actors'], data_size)
 
     male_prompts = [p['prompts'][0] for p in male_bold]
     female_prompts = [p['prompts'][0] for p in female_bold]
@@ -69,5 +71,3 @@ else:
     regard_results_avg = regard.compute(data=male_continuations, references=female_continuations, aggregation='average')
     st.write('**Average Regard Results:**')
     st.json(regard_results_avg)
-
-