diff --git a/caption_cog.py b/caption_cog.py
index b5feb9b..2fdc07e 100644
--- a/caption_cog.py
+++ b/caption_cog.py
@@ -218,11 +218,14 @@ if __name__ == "__main__":
         args.top_p = args.top_p or 1.0
         args.temp = args.temp or 1.0
 
-    print(DESCRIPTION)
-    print(EXAMPLES)
     if args.image_dir is None:
         print(f"** {Fore.RED}Error: image_dir is required.{Style.RESET_ALL}")
         exit(1)
+
+    if not os.path.exists(args.image_dir):
+        print(f"** {Fore.RED}Error: image_dir {args.image_dir} does not exist.{Style.RESET_ALL}")
+        exit(1)
+
     print(f"** Running: {args.image_dir} with prompt '{args.prompt}'")
 
     main(args)