Update README.md

README.md

@@ -65,17 +65,15 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 
 base_model_id = "meta-llama/Llama-3.2-1B-Instruct"
-adapter_id = "YOUR_USERNAME/llama32-1b-python-docstrings-qlora"
+adapter_id = "Abdul1102/llama32-1b-python-docstrings-qlora"
 
 tokenizer = AutoTokenizer.from_pretrained(base_model_id)
 model = AutoModelForCausalLM.from_pretrained(base_model_id, device_map="auto")
 model = PeftModel.from_pretrained(model, adapter_id)
 
 def make_prompt(code: str) -> str:
-    return (
-        "Write a one-line Python docstring for this function:\n\n"
-        f"{code}\n\n\"\"\""
-    )
+    return
+        f'Write a one-line Python docstring for this function:\n\n{code}\n\n"""'
 
 code = "def add(a, b):\n    return a + b"
 inputs = tokenizer(make_prompt(code), return_tensors="pt").to(model.device)