Upload folder using huggingface_hub

README.md

@@ -16,6 +16,6 @@ Fine-tuned Qwen2.5-Coder-1.5B on Python code instructions.
 **Model:** Qwen/Qwen2.5-Coder-1.5B-Instruct | **Fine-tune:** LoRA (r=8)
 
 ## Links
-- **Trained model:** [arinbalyan/starcoder2-code-lm](https://huggingface.co/arinbalyan/starcoder2-code-lm)
+- **Trained model:** [arinbalyan/qwen-coder-python](https://huggingface.co/arinbalyan/qwen-coder-python)
 - **Training notebook:** [Kaggle](https://www.kaggle.com/code/arinbalyan/fine-tune-starcoder2-1b-for-python-code)
 - **HF Profile:** [arinbalyan](https://huggingface.co/arinbalyan)

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-MODEL_ID = "arinbalyan/starcoder2-code-lm"
+MODEL_ID = "arinbalyan/qwen-coder-python"
 
 theme = (
     gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")