Upload folder using huggingface_hub

README.md

@@ -1,13 +1,21 @@
 ---
-title: Codellm Python
-emoji: 🏆
+title: CodeLM
+emoji: 🐍
 colorFrom: blue
-colorTo: red
+colorTo: green
 sdk: gradio
 sdk_version: 6.19.0
-python_version: '3.13'
 app_file: app.py
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# CodeLM — Python Code Generation
+
+Fine-tuned StarCoder2-1B on Python code instructions.
+
+**Base:** bigcode/starcoder2-1B | **Fine-tune:** LoRA (r=8)
+
+## Links
+- **Trained model:** [arinbalyan/starcoder2-code-lm](https://huggingface.co/arinbalyan/starcoder2-code-lm)
+- **Training notebook:** [Kaggle](https://www.kaggle.com/code/arinbalyan/fine-tune-starcoder2-python-code)
+- **HF Profile:** [arinbalyan](https://huggingface.co/arinbalyan)

app.py

@@ -0,0 +1,58 @@
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+MODEL_ID = "arinbalyan/starcoder-code-lm"
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+model = AutoModelForCausalLM.from_pretrained(
+    MODEL_ID,
+    dtype=torch.float16,
+    device_map="auto",
+)
+
+def generate_code(instruction):
+    prompt = f"### Instruction:\n{instruction}\n\n### Python Code:\n"
+    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
+    outputs = model.generate(
+        **inputs,
+        max_new_tokens=256,
+        temperature=0.7,
+        do_sample=True,
+        pad_token_id=tokenizer.eos_token_id,
+    )
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return response.split("### Python Code:\n")[-1].strip()
+
+with gr.Blocks(title="CodeLM - Python Code Generation") as demo:
+    gr.Markdown("""
+    # CodeLM - Python Code Generation
+    
+    Fine-tuned StarCoder2-1B with LoRA on Python code instructions.
+    
+    Enter a description of what you want to code, and the model will generate Python code.
+    """)
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            instruction_input = gr.Textbox(
+                label="What do you want to code?",
+                placeholder="e.g., Write a function to check if a string is a palindrome",
+                lines=3
+            )
+            generate_btn = gr.Button("Generate Code", variant="primary")
+        
+        with gr.Column(scale=3):
+            code_output = gr.Code(
+                label="Generated Python Code",
+                language="python",
+                lines=20
+            )
+    
+    generate_btn.click(
+        fn=generate_code,
+        inputs=instruction_input,
+        outputs=code_output
+    )
+
+demo.launch(theme=gr.themes.Soft())

requirements.txt

@@ -0,0 +1,3 @@
+transformers>=4.40
+torch>=2.0
+accelerate>=1.0