Update README.md

README.md

@@ -34,9 +34,9 @@ prompt.
 
 ```python
 >>> from transformers import AutoTokenizer, AutoModelForCausalLM
->>> model_name = "Sharathhebbar24/Instruct_GPT"
+>>> model_name = "Sharathhebbar24/Instruct_GPT_v1"
 >>> model = AutoModelForCausalLM.from_pretrained(model_name)
->>> tokenizer = AutoTokenizer.from_pretrained("gpt2-medium")
+>>> tokenizer = AutoTokenizer.from_pretrained(model_name)
 >>> def generate_text(prompt):
 >>>  inputs = tokenizer.encode(prompt, return_tensors='pt')
 >>>  outputs = model.generate(inputs, max_length=64, pad_token_id=tokenizer.eos_token_id)