Update post_generator.py

post_generator.py

@@ -41,7 +41,7 @@ def get_prompt(few_shot_instance, length, language, tag):
         prompt += "4) Use the writing style as per the following examples."
 
     for i, post in enumerate(examples):
-        post_text = post['text_blocks']
+        post_text = post['text']
         prompt += f'\n\n Example {i+1}: \n\n {post_text}'
 
         if i == 1:  # Use max two samples