From e409c120d08f52738b67c3a6833cc36482f4b788 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Fri, 8 Jul 2022 06:01:07 -0400
Subject: [PATCH] update readme

---
 README.md | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 3729e75..7e7a637 100644
--- a/README.md
+++ b/README.md
@@ -68,15 +68,14 @@ images = model.generate_images(
 )
 ```
 
-Note: you will have to move the images to the cpu and convert to numpy, e.g. 
+To get an image into PIL format you will have to first move the images to the CPU and convert the tensor to a numpy array.
 ```python
 images = images.to('cpu').detach().numpy()
 ```
-Then image $i$ can be coverted to a PIL.Image 
+Then image $i$ can be coverted to a PIL.Image and saved with its `save` method
 ```python
 image = Image.fromarray(images[i])
 ```
-and saved with its `save` method `image.save('image.png')`.
 
 ### Interactive