Update README.md

README.md

@@ -18,7 +18,7 @@ inference:
     max_length: 120
     do_sample: True
     temperature: 0.8
----
+--\
 
 # GPT-3 small
 
@@ -28,8 +28,8 @@ Pretrained GPT-3 small, continuing the development of GPT NEO, with architecture
 ~~~~
 from transformers import GPT2Tokenizer, GPTNeoForCausalLM
 
-tokenizer = GPT2Tokenizer.from_pretrained('gpt3-small-finetune-cnndaily-news')
-model = GPTNeoForCausalLM.from_pretrained('gpt3-small-finetune-cnndaily-news')
+tokenizer = GPT2Tokenizer.from_pretrained('minhtoan/gpt3-small-finetune-cnndaily-news')
+model = GPTNeoForCausalLM.from_pretrained('minhtoan/gpt3-small-finetune-cnndaily-news')
 
 text = "Ever noticed how plane seats appear to be getting smaller and smaller? "
 input_ids = tokenizer.encode(text, return_tensors='pt')