Update README.md

README.md

@@ -31,7 +31,7 @@ tokenizer = AutoTokenizer.from_pretrained('/path/to/tokenizer')
 model = GPT2LMHeadModel.from_pretrained('/path/to/output').to(device)
 input_ids = tokenizer.encode(enzyme_class,return_tensors='pt').to(device)
 # change max_length or num_return_sequences to your requirements
-output = model.generate(input_ids, top_k=8, repetition_penalty=1.2, max_length=1024,
+output = model.generate(input_ids, top_k=9, repetition_penalty=1.2, max_length=1024,
                         eos_token_id=1,pad_token_id=0,do_sample=True, num_return_sequences=100)
 ```