Update README.md

README.md

@@ -47,8 +47,8 @@ CREATE TABLE "table_name" (
 
 ```python
 import torch
-from modelscope import snapshot_download, AutoModelForCausalLM, AutoTokenizer,GenerationConfig
-model_dir = snapshot_download("iic/NL2SQL-StarCoder-15B")
+from transformers import AutoModelForCausalLM, AutoTokenizer
+model_dir = "gabrielpondc/NL2SQL-StarCoder-15B"
 tokenizer = AutoTokenizer.from_pretrained(model_dir, device_map="auto", 
                               trust_remote_code=True, torch_dtype=torch.float16)
 tokenizer.padding_side = "left"