Update README.md

README.md

@@ -15,14 +15,35 @@ metrics:
 - accuracy
 ---
 
-
 # phi-2-basic-maths
 
 This model is a fine-tuned version of [microsoft/phi-2](https://huggingface.co/microsoft/phi-2) on an [GSM8K dataset](https://huggingface.co/datasets/gsm8k).
 
-## Model description
+## Model Description
+
+The objective of this model is to evaluate Phi-2's ability to provide correct solutions to reasoning problems after fine-tuning. This model was trained using techniques such as TRL, LoRA quantization, and Flash Attention.
+
+To test it, you can use the following code:
+
+```python
+import torch
+from peft import AutoPeftModelForCausalLM
+from transformers import AutoTokenizer, pipeline
+
+# Specify the model ID
+peft_model_id = "Menouar/phi-2-basic-maths"
+
+# Load Model with PEFT adapter
+model = AutoPeftModelForCausalLM.from_pretrained(
+  peft_model_id,
+  device_map="auto",
+  torch_dtype=torch.float16
+)
+
+tokenizer = AutoTokenizer.from_pretrained(peft_model_id)
 
-More information needed
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
+```
 
 ## Training procedure