Update README.md

README.md

@@ -76,9 +76,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 prompt = "//module half adder "
 device='cuda'
 # Load model and tokenizer
-model_name = "shailja/CodeGen_16B_Verilog"
-tokenizer = AutoTokenizer.from_pretrained("shailja/fine-tuned-codegen-16B-Verilog")
-model = AutoModelForCausalLM.from_pretrained("shailja/fine-tuned-codegen-16B-Verilog").to(device)
+model_name = "shailja/fine-tuned-codegen-16B-Verilog"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 
 # Sample
 input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)