Update README.md

README.md

@@ -16,8 +16,8 @@ import torch
 import torch.nn.functional as F
 
 # load model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("hugohrban/progen2-small-mix7", trust_remote_code=True)
-tokenizer = Tokenizer.from_pretrained("hugohrban/progen2-small-mix7")
+model = AutoModelForCausalLM.from_pretrained("hugohrban/progen2-small", trust_remote_code=True)
+tokenizer = Tokenizer.from_pretrained("hugohrban/progen2-small")
 tokenizer.no_padding()
 
 # prepare input