From 5d31ad51490766acd5d4f8be1cc8b1fbdddd27be Mon Sep 17 00:00:00 2001
From: Michael Foiani <sotech117@michaels-mbp-3.devices.brown.edu>
Date: Sat, 7 May 2022 17:58:34 -0400
Subject: mege with pycache

---
 main.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'main.py')

diff --git a/main.py b/main.py
index 4497cce0..193d7c7a 100644
--- a/main.py
+++ b/main.py
@@ -70,7 +70,10 @@ def main():
     
     final_image = tf.squeeze(my_model.x)
 
-    plt.imshow(final_image).astype('uint8')
+    # convert image to uint8
+    final_image = tf.cast(final_image, tf.uint8)
+
+    plt.imshow(final_image)
 
     imsave(ARGS.savefile, final_image)
 
-- 
cgit v1.2.3-70-g09d2