Update README.md

README.md

@@ -66,8 +66,10 @@ def generate_response(messages, max_new_tokens=1024, temperature=0.1):
 
 1. **Translation Tasks**
 
-```bash
+```python
+system_prompt = "You're a Wolof AI assistant. Please always provide detailed and useful answers to the user queries."
 messages = [
+    {"role": "system", "content": system_prompt},
     {"role": "user", "content": "Translate to Wolof: Bassirou Diomaye Faye is the new Senegalese president. He is 44 years old"}
 ]
 print(generate_response(messages))
@@ -76,7 +78,9 @@ print(generate_response(messages))
 2. **Code generation**
 
 ```python
+system_prompt = "You're a Wolof AI assistant. Please always provide detailed and useful answers to the user queries"
 messages = [
+    {"role": "system", "content": system_prompt},
     {"role": "user", "content": "Bindal ab klaas Python buy wone ni ñuy jëfandikoo dataframe yi ci Pandas"}
 ]
 print(generate_response(messages))
@@ -84,8 +88,8 @@ print(generate_response(messages))
 
 3. **Problem Solving**
 
-```bash
-system_prompt = "You're a Wolof AI assistant. Please always provide detailed and useful answers to the user queries"
+```python
+system_prompt = "You're a Wolof AI assistant. Please always provide detailed and useful answers to the user queries."
 messages = [
     {"role": "system", "content": system_prompt},
     {"role": "user", "content": "Ndax nga mën ma won ni ñuy resolver problème bii: Fatou dafa jënd 3 kilo ceeb, 2 kilo diw ak 5 kilo sukër. Ceeb gi wenn kilo 500 CFA la, diw gi 1200 CFA kilo bi, sukër gi 750 CFA kilo bi. Ñaata la wara fay?"}
@@ -99,7 +103,6 @@ pprint(generate_response(messages))
 
 ```python
 system_prompt = "You are a skilled Wolof storyteller (Gewël) with deep knowledge of African folktales and traditions. Write engaging stories in Wolof that reflect African cultural values and wisdom."
-
 messages = [
     {"role": "system", "content": system_prompt},
     {"role": "user", "content": "Bindal ab léeb ci gaynde gi lekk muus mi"}