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)
+completion = model.generate(input_ids, max_length=100)
 print(tokenizer.decode(completion[len(input_ids):])[0])
 ```