Update README.md

README.md

@@ -10,14 +10,17 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel
 
-tokenizer = AutoTokenizer.from_pretrained("alpecevit/Qwen2.5-Coder-7B-Instruct-text2sql")
+
+MODEL_NAME = "alpecevit/Qwen2.5-Coder-7B-Instruct-text2sql"
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(
-    "alpecevit/Qwen2.5-Coder-7B-Instruct-text2sql",
+    MODEL_NAME,
     torch_dtype="auto",
     device_map="auto"
 )
 
-model = PeftModel.from_pretrained(model, LORA_WEIGHTS)
+model = PeftModel.from_pretrained(model, MODEL_NAME)
 model.eval()
 
 test_data = {