Update README.md

README.md

@@ -56,8 +56,8 @@ def generate_prompt(description, inputs, outputs):
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # load model and tokenizer
-model = AutoModelForCausalLM.from_pretrained(args.model_path, device_map="auto")
-tokenizer = AutoTokenizer.from_pretrained(args.model_path)
+model = AutoModelForCausalLM.from_pretrained("iamtarun/pycompetitive-codegen350M-qlora", device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained("iamtarun/pycompetitive-codegen350M-qlora")
 
 # loading model for inference
 model.eval()