Final fix for README

README.md

@@ -57,27 +57,27 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from safetensors.torch import load_file
 
 \# Load tokenizer and model
+```python
+tokenizer = AutoTokenizer.from_pretrained("aryan27/geometry-model-hf")
 
-tokenizer =
-AutoTokenizer.from_pretrained(\"your-username/minillm-geometry-engine\")
-
-model =
-AutoModelForCausalLM.from_pretrained(\"your-username/minillm-geometry-engine\")
+model = AutoModelForCausalLM.from_pretrained("aryan27/geometry-model-hf")
+```
 
 \# Example prompt
+```python
+prompt = "Draw a point A at 3,4"
 
-prompt = \"Draw a point A at 3,4\"
-
-inputs = tokenizer(prompt, return_tensors=\"pt\")
+inputs = tokenizer(prompt, return_tensors="pt")
 
-outputs = model.generate(\*\*inputs, max_length=200)
+outputs = model.generate(*inputs, max_length=200)
 
-print(tokenizer.decode(outputs\[0\], skip_special_tokens=True))
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
 
 Install required libraries:
-
+```shell
 pip install safetensors transformers
-
+```
 ## **Training Details**
 
 -   **Dataset**: Custom Geometry CoT dataset with 10,000 geometry problems and solutions