more examples + ease of use fixes

1 files changed, 1 insertions, 1 deletions

diff --git a/main.py b/main.py
index e60f14d1..3649309a 100644
--- a/main.py
+++ b/main.py
@@ -10,7 +10,6 @@ from model import YourModel
 
 from skimage.io import imread, imsave
 from matplotlib import pyplot as plt
-import numpy as np
 
 
 def parse_args():
@@ -105,6 +104,7 @@ def main():
 
     # 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()