Logs and cleaning.

inference_manga_v2.py

@@ -55,13 +55,12 @@ for root, dirs, files in os.walk(base, topdown=True):
         img_LR = img.unsqueeze(0)
         img_LR = img_LR.to(device)
 
-        #print(f"Start upscaling tile {x}...")
+        print(f"Start upscaling tile {x}...")
         with torch.no_grad():
             output = model(img_LR).squeeze(dim=0).float().cpu().clamp_(0, 1).numpy()
         output = np.transpose(output, (1, 2, 0))
         output = (output * 255.0).round()
-        #print(f"Finished upscaling tile {x}, saving tile.")
-        #print(output_dir)
+        print(f"Finished upscaling tile {x}, saving tile.")
         cv2.imwrite(file_path, output)
 
 # Join all tiles

process_image.py

@@ -36,7 +36,7 @@ def inference(img, size, type):
     #img_out = Image.open(img_out_path)
 
     # Remove input and output image
-    #run_cmd(f"rm -f {img_in_path}")
+    run_cmd(f"rm -rf {INPUT_DIR}")
 
     img_out.thumbnail((600, 600), Image.ANTIALIAS)