Update README.md

README.md

@@ -43,7 +43,7 @@ def generate_prompt(text: str) -> str:
 text = '你觉得如果有一天猫咪统治地球会是什么样子？会有哪些新的法律或者规定出台？'
 prompt = generate_prompt(text)
 inputs = tokenizer.encode(prompt, return_tensors='pt').to(device)
-response = model.generate(inputs)
+response = model.generate(inputs, max_new_tokens=400)
 outputs = tokenizer.batch_decode(response)
 outputs = outputs[0].replace(text, '').strip().strip('</s>')
 print(f'Meow: {outputs}')