Update README.md

README.md

@@ -58,7 +58,49 @@ This model is a fine-tuned version of Qwen3-1.7B using ORPO (Odds Ratio Preferen
 Use the code below to get started with the model.
 
 ```python
+from huggingface_hub import login
+from transformers import AutoTokenizer, AutoModelForCausalLM
+from peft import PeftModel
 
+login(token="")
+
+tokenizer = AutoTokenizer.from_pretrained("unsloth/Qwen3-1.7B",)
+base_model = AutoModelForCausalLM.from_pretrained(
+    "unsloth/Qwen3-1.7B",
+    device_map={"": 0}, token=""
+)
+
+model = PeftModel.from_pretrained(base_model,"Rustamshry/datascience-RLHF")
+
+
+prompt = """Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
+
+### Instruction:
+{}
+
+### Input:
+{}
+
+### Response:
+{}"""
+
+
+inputs = tokenizer(
+    [
+        prompt.format(
+            "You are an AI assistant that helps people find information",
+            "What is the k-Means Clustering algorithm and what is it's purpose?", 
+            "",  
+        )
+    ],
+    return_tensors="pt",
+).to("cuda")
+
+
+from transformers import TextStreamer
+
+text_streamer = TextStreamer(tokenizer)
+_ = model.generate(**inputs, streamer=text_streamer, max_new_tokens=1800)
 ```