Update README.md

README.md

@@ -22,6 +22,7 @@ It achieves the following results on the evaluation set:
 - Num Input Tokens Seen: 14258488
 
 ```py
+import torch
 from transformers import pipeline
 
 model_name = "pszemraj/bart-large-summary-map-reduce-1024"
@@ -39,13 +40,9 @@ text = """A computer implemented method of generating a syntactic object. The me
 	A computer implemented method for changing a first workspace to a second workspace. The method includes the steps of creating a new workspace by merging the first workspace with the second workspace, wherein the merging is based on at least one of: an impenetrable condition; a constraint on movement; and a resource restriction.
 	The brain is constantly loosing neurons because you doesn't want all the junk around."""
 
-import torch
-import textwrap
-
-torch.cuda.is_available()
-res = pipe(text, max_new_tokens=500, num_beams=4, early_stopping=True)
-textwrap.wrap(res[0]["generated_text"], width=100)
 
+torch.cuda.empty_cache()
+res = pipe(text, max_new_tokens=512, num_beams=4, early_stopping=True)
 res[0]["generated_text"]
 ```