Update README.md

README.md

@@ -1,3 +1,35 @@
----
-license: mit
----
+---
+license: mit
+---
+# 🧠 CodeGen-Alpaca-1B
+
+This is a fine-tuned version of `StarCoderBase-1B` using the **CodeAlpaca (2k subset)** dataset.  
+It has been trained with **QLoRA** on Google Colab for lightweight, memory-efficient code generation.
+
+---
+
+## ✨ Model Capabilities
+
+This model understands **instruction-style prompts** for generating code in multiple programming languages, especially Python.
+
+---
+
+## 📦 How to Use
+
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+model_id = "key-life/codegen-alpaca-1b"
+
+# Load model & tokenizer
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
+
+# Example prompt
+prompt = "### Instruction:\nWrite a Python function to check if a number is prime.\n\n### Response:\n"
+inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
+
+# Generate code
+outputs = model.generate(**inputs, max_new_tokens=128)
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))