Update README.md

README.md

@@ -22,7 +22,6 @@ This is a model capable of injecting bugs into correct Python programs. It was u
 You may use the model in the following way:
 
 ```python
-# %%
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
@@ -70,7 +69,10 @@ def bitcount(n):
 
 corrupted_programs = corrupt(
     test_code, model, tokenizer, 
-    do_sample=True, temperature=0.5, max_new_tokens=4096, num_return_sequences=2,
+    do_sample=True, 
+    temperature=0.5, 
+    max_new_tokens=4096, 
+    num_return_sequences=5,
 )
 
 for corrupted_program in corrupted_programs: