Update README.md

README.md

@@ -28,7 +28,7 @@ For example, your prompt can look like
 instruction = "Write a function to compute the GCD between two integers a and b"
 prompt = f"Question:{instruction}\n\nAnswer:"
 input_ids = tokenizer(prompt, return_tensors="pt")["input_ids"]
-completion = model.generate(input_ids, max_length=100)
-print(tokenizer.decode(completion[len(input_ids):])[0])
+completion = model.generate(input_ids, max_length=200)
+print(tokenizer.batch_decode(completion[:,input_ids.shape[1]:])[0])
 ```