Update README.md

README.md

@@ -21,9 +21,9 @@ import torch
 import re
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-config = PeftConfig.from_pretrained("Ashishkr/llama2-qrecc-context-resolution")
+config = PeftConfig.from_pretrained("Ashishkr/llama2-qrecc")
 model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf")
-model = PeftModel.from_pretrained(model, "Ashishkr/llama2-qrecc-context-resolution").to(device)
+model = PeftModel.from_pretrained(model, "Ashishkr/llama2-qrecc").to(device)
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
 
 def response_generate(