Update README.md

README.md

@@ -17,8 +17,11 @@ This model is a fine-tuned version of [facebook/opt-350m](https://huggingface.co
 
 ## Model description
 
+Make sure to set max_new_tokens = 20; otherwise, the model will generate one token at a time.
+
 nlp = pipeline("text-generation",
-                model="piazzola/tmp_trainer")
+                model="piazzola/tmp_trainer",
+                max_new_tokens=20)
                 
 nlp("Landlord hereby demises unto Tenant, and Tenant hereby leases from Landlord for the terms and upon the conditions set forth in this Lease 30,600 square feet of space in the building located at 15 Firstfield Road, Gaithersburg, Maryland, the Building, as set forth on Exhibit A, hereto attached, said space being referred to as the Premises.")