Update README.md

README.md

@@ -33,6 +33,7 @@ The model was instruction tuned on the dataset in the following way:
 
 Given the set of promts: 
 
+``` python
 prompts = [
     "Write a creative short story based on the following title:",
     "Here is a title for a story. Craft a short narrative around it:",
@@ -40,10 +41,13 @@ prompts = [
     "Imagine a short story that starts with this title:",
     "Create a brief story with the following title:"
 ]
+```
 
 each training example is generated by concatenating one of the prompts with the 'title' and 'selftext' in the following way:
 
-    concatenated_texts = [random.choice(prompts) + " " + title + "</s>" + "Story: " + selftext for title, selftext in zip(titles, selftexts)]
+``` python
+concatenated_texts = [random.choice(prompts) + " " + title + "</s>" + "Story: " + selftext for title, selftext in zip(titles, selftexts)]
+```
 
 ### Training hyperparameters