Update README.md

README.md

@@ -24,7 +24,7 @@ sentence = "i left the keys in the car."
 
 with torch.no_grad():
     inputs = tokenizer([sentence], return_tensors="pt")
-    outputs = model.generate(**inputs)
+    outputs = model.generate(**inputs, max_new_tokens=100)
     generated_text = tokenizer.decode(outputs[0])
 
 print(generated_text)