Update README.md

README.md

@@ -97,8 +97,21 @@ def generate_text(model, tokenizer, prompt,
     generated_text = generated_text.split(prompt)[-1].strip()
 
     return generated_text
+```
+
+## Inference Example
+
+This section showcases how to use the model for inference.
 
+**User Query:**
+
+```
 user_query = "I'm a 35-year-old male experiencing symptoms like fatigue, increased sensitivity to cold, and dry, itchy skin. Could these be indicative of hypothyroidism?"
+
+```
+**Answer:** 
+
+```
 generated_text = generate_text(model, tokenizer, user_query)    
 print(generated_text)