Update README.md

README.md

@@ -2,27 +2,88 @@
 language:
 - en
 license: apache-2.0
-license_name: apache-2.0
-license_link: https://www.apache.org/licenses/LICENSE-2.0
 library_name: transformers
 tags:
 - code-generation
 - python
-- qwen
-- fine-tuned
-- lora
 - programming
+- fine-tuned
 pipeline_tag: text-generation
-datasets:
-- CodeAlpaca-20k
-- mbpp
-- HumanEval
-base_model: Qwen/Qwen2.5-7B-Instruct
 widget:
-- text: "### Instruction:\nWrite a Python function to reverse a string\n\n### Input:\n\n\n### Response:"
-  example_title: "Reverse String"
-- text: "### Instruction:\nImplement binary search algorithm\n\n### Input:\nReturn -1 if element not found\n\n### Response:"
-  example_title: "Binary Search"
-- text: "### Instruction:\nCreate a class for a Stack\n\n### Input:\nInclude push, pop, and is_empty methods\n\n### Response:"
-  example_title: "Stack Class"
----
+- text: |-
+    ### Instruction:
+    Write a Python function to reverse a string
+
+    ### Input:
+
+
+    ### Response:
+  example_title: Reverse String
+- text: |-
+    ### Instruction:
+    how are you?
+
+    ### Input:
+
+
+    ### Response:
+  example_title: Non-Code Request
+- text: |-
+    ### Instruction:
+    Implement quicksort algorithm in Python
+
+    ### Input:
+    Use recursion and list comprehensions
+
+    ### Response:
+  example_title: Quicksort Algorithm
+base_model: Mercy-62/python-code-only-Qwen-lora
+datasets:
+- sahil2801/CodeAlpaca-20k
+- google-research-datasets/mbpp
+- openai/openai_humaneval
+---
+
+# 🐍 Python Code-Only Qwen
+
+**A fine-tuned model that generates ONLY executable Python code, with no explanations or conversations.**
+
+## 🚀 Quick Inference Example
+
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# Load model
+model = AutoModelForCausalLM.from_pretrained("Mercy-62/python-code-only-Qwen-lora")
+tokenizer = AutoTokenizer.from_pretrained("Mercy-62/python-code-only-Qwen-lora")
+
+# Use exact same prompt format from training
+alpaca_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:
+{}"""
+
+# Enable faster inference (if using Unsloth)
+# FastLanguageModel.for_inference(model)
+
+# Test 1: Simple code generation
+inputs = tokenizer(
+    [
+        alpaca_prompt.format(
+            "Write a Python function to reverse a string",  # instruction
+            "",  # input (leave empty if no context)
+            "",  # output - leave blank for generation
+        )
+    ],
+    return_tensors="pt"
+).to("cuda")
+
+outputs = model.generate(**inputs, max_new_tokens=128, use_cache=True)
+print("Test 1 - Reverse string function:")
+print(tokenizer.batch_decode(outputs)[0])