From b1b065d5c3ca405c42dbe26fc8f468328d828436 Mon Sep 17 00:00:00 2001 From: Niles Rogoff Date: Fri, 10 Feb 2017 18:40:39 -0500 Subject: [PATCH] Fix small bug with argument counting --- upscale.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/upscale.py b/upscale.py index d69f44f..7da6a82 100755 --- a/upscale.py +++ b/upscale.py @@ -5,7 +5,7 @@ null = open(os.devnull, "w") infile = sys.argv[1] outfile = ".".join(sys.argv[1].split(".")[:-1]) + "-waifu2x.png" noise_level = False -if len(sys.argv) == 2: +if len(sys.argv) == 3: if sys.argv[2] in ["0", "1", "2"]: print("Assuming you don't want to write to a file named " + sys.argv[2]) noise_level = int(sys.argv[2])