Zekun Wu commited on
Commit
d8c9fb0
·
1 Parent(s): cd39b1e
Files changed (1) hide show
  1. pages/1_Demo_1.py +6 -0
pages/1_Demo_1.py CHANGED
@@ -83,9 +83,15 @@ else:
83
  if st.session_state.get('male_continuations') and st.session_state.get('female_continuations'):
84
  st.subheader('Step 3: Sample Generated Texts')
85
 
 
86
  samples_df = pd.DataFrame({
87
  'Male Prompt': st.session_state['male_prompts'],
88
  'Male Continuation': st.session_state['male_continuations'],
 
 
 
 
 
89
  'Female Prompt': st.session_state['female_prompts'],
90
  'Female Continuation': st.session_state['female_continuations']
91
  })
 
83
  if st.session_state.get('male_continuations') and st.session_state.get('female_continuations'):
84
  st.subheader('Step 3: Sample Generated Texts')
85
 
86
+ st.write("Male Data Samples:")
87
  samples_df = pd.DataFrame({
88
  'Male Prompt': st.session_state['male_prompts'],
89
  'Male Continuation': st.session_state['male_continuations'],
90
+ })
91
+ st.write(samples_df)
92
+
93
+ st.write("Female Data Samples:")
94
+ samples_df = pd.DataFrame({
95
  'Female Prompt': st.session_state['female_prompts'],
96
  'Female Continuation': st.session_state['female_continuations']
97
  })