update

pages/1_Demo_1.py

@@ -44,15 +44,15 @@ else:
     GPT2 = gpt2()
 
     st.write('Generating text for male prompts...')
-    male_generation = GPT2.generate_text(male_prompts, pad_token_id=50256, max_length=50, do_sample=False,truncation=True)
+    male_generation = GPT2.text_generation(male_prompts, pad_token_id=50256, max_length=50, do_sample=False,truncation=True)
     print(male_generation)
-    male_continuations = [gen.replace(prompt, '') for gen, prompt in zip(male_generation, male_prompts)]
+    male_continuations = [gen[0]['generated_text'].replace(prompt, '') for gen, prompt in zip(male_generation, male_prompts)]
 
     st.write('Generating text for female prompts...')
 
-    female_generation = GPT2.generate_text(female_prompts, pad_token_id=50256, max_length=50, do_sample=False,truncation=True)
+    female_generation = GPT2.text_generation(female_prompts, pad_token_id=50256, max_length=50, do_sample=False,truncation=True)
     print(male_generation)
-    female_continuations = [gen.replace(prompt, '') for gen, prompt in zip(female_generation, female_prompts)]
+    female_continuations = [gen[0]['generated_text'].replace(prompt, '') for gen, prompt in zip(male_generation, male_prompts)]
 
     st.write('Generated {} male continuations'.format(len(male_continuations)))
     st.write('Generated {} female continuations'.format(len(female_continuations)))