Update app.py

app.py

@@ -110,9 +110,11 @@ if task_choice == "Data Generation":
     if few_shot_examples:
         system_prompt += "Use the following few-shot examples as a reference:\n"
         for example in few_shot_examples:
-            system_prompt += f"Example: {example['content']}, Label: {example['label']}\n"
+            system_prompt += f"Example: {example['content']}, Sentiment: {example['label']}\n"
     system_prompt += f"Each example should have between {min_words} and {max_words} words.\n"
     system_prompt += "Think step by step while generating the examples."
+    ####
+    system_prompt += f"Generate {example['num_to_generate']}"
 
     st.write("System Prompt:")
     st.code(system_prompt)