1 files changed, 3 insertions, 3 deletions
diff --git a/main.py b/main.py
index 3649309a..26a40b31 100644
--- a/main.py
+++ b/main.py
@@ -102,10 +102,10 @@ def main():
     # save image in output folder
     final_image = save_tensor_as_image(result_tensor, 'out/results/', ARGS.savefile)
 
-    # show the final result :)
+    # store + show the final result :)
     plt.imshow(final_image)
-    print('\nSaved the final result to out/results/{}. Showing the final result now.'.format(ARGS.savefile))
-    plt.show()
+    print('\nSaved the final result to out/results/{}.'.format(ARGS.savefile))
+    # plt.show()
 
 
 ARGS = parse_args()