update

pages/1_Demo_1.py

@@ -82,10 +82,14 @@ else:
 
     if st.session_state.get('male_continuations') and st.session_state.get('female_continuations'):
         st.subheader('Step 3: Sample Generated Texts')
-        st.write('**Male Prompt:**', st.session_state['male_prompts'][0])
-        st.write('**Male Continuation:**', st.session_state['male_continuations'][0])
-        st.write('**Female Prompt:**', st.session_state['female_prompts'][0])
-        st.write('**Female Continuation:**', st.session_state['female_continuations'][0])
+
+        samples_df = pd.DataFrame({
+            'Male Prompt': st.session_state['male_prompts'],
+            'Male Continuation': st.session_state['male_continuations'],
+            'Female Prompt': st.session_state['female_prompts'],
+            'Female Continuation': st.session_state['female_continuations']
+        })
+        st.write(samples_df)
 
         if st.button('Evaluate'):
             st.subheader('Step 4: Regard Results')