Update README.md

README.md

@@ -101,8 +101,8 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
 # Load MyanmarGPT-Chat model and tokenizer
-model = GPT2LMHeadModel.from_pretrained("jojo-ai-mst/MyanmarGPT")
-tokenizer = GPT2Tokenizer.from_pretrained("jojo-ai-mst/MyanmarGPT")
+model = GPT2LMHeadModel.from_pretrained("jojo-ai-mst/MyanmarGPT-Chat")
+tokenizer = GPT2Tokenizer.from_pretrained("jojo-ai-mst/MyanmarGPT-Chat")
 
 def generate_text(prompt, max_length=300, temperature=0.8, top_k=50):
     input_ids = tokenizer.encode(prompt, return_tensors="pt").cuda() # remove .cude() if only cpu