Update README.md

README.md

@@ -20,7 +20,7 @@ It has been trained using [TRL](https://github.com/huggingface/trl).
 import torch
 from transformers import pipeline, AutoTokenizer
 
-question = "Write a Python function that takes a list of numbers and returns the list sorted in ascending order without using the built-in `sorted()` function."
+question = "Write a Python function that takes a list of numbers and returns the list sorted in ascending order without using the built-in `sorted()` function. Return ONLY code in your output. ###Python code:"
 generator = pipeline("text-generation", model="mariasandu/python-coding-assistant-v2", device="cuda")
 
 tokenizer = AutoTokenizer.from_pretrained("mariasandu/python-coding-assistant-v2")