Update README.md

README.md

@@ -76,7 +76,20 @@ deci_generator = pipeline("text-generation",
 )
 
 prompt = "How do I make the most delicious pancakes the world has ever tasted?"
-response = deci_generator(prompt)[0]['generated_text']
+
+SYSTEM_PROMPT_TEMPLATE ="""
+### System:
+You are an AI assistant that follows instruction extremely well. Help as much as you can.
+### User:
+{instruction}
+### Assistant:
+"""
+
+# Function to construct the prompt using the new system prompt template
+def get_prompt_with_template(message: str) -> str:
+    return SYSTEM_PROMPT_TEMPLATE.format(instruction=message)
+
+response = deci_generator(get_prompt_with_template(prompt))[0]['generated_text']
 print(response)
 ```