Update README.md

README.md

@@ -108,15 +108,17 @@ print(tokenizer.decode(input_ids[0]))
 
 <details>
 
-<summary>Output</summary>
+<summary>Prompt</summary>
 
-**Prompt**: 
 ```python
 """
 <s> [INST] Tell me a recipe for a mai tai. [/INST]
 """
 ```
 
+</details>
+
+
 ```python
 print("\n\n*** Generate:")
 with torch.autocast("cuda", dtype=torch.bfloat16):
@@ -140,7 +142,10 @@ print(response)
 
 ```
 
-**Generation**:
+<details>
+
+<summary>Generation</summary>
+
 ```python
 """1 oz light rum
 ½ oz dark rum