update

pages/1_Demo_1.py

@@ -60,6 +60,8 @@ else:
             st.write('Generating text for male prompts...')
             male_generation = GPT2.text_generation(st.session_state['male_prompts'], pad_token_id=50256, max_length=50,
                                                    do_sample=False, truncation=True)
+
+            print(male_generation)
             st.session_state['male_continuations'] = [gen['generated_text'].replace(prompt, '') for gen, prompt in
                                                       zip(male_generation, st.session_state['male_prompts'])]