Update README.md

README.md

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