Add logs.

app.py

@@ -23,7 +23,7 @@ def inference(img):
     run_cmd("mkdir " + OUTPUT_DIR)
     img.save(INPUT_DIR + "1.jpg", "JPEG")
 
-    os.system("python test.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
+    run_cmd("python test.py "+ os.path.join(INPUT_DIR, "1.jpg") + " " + os.path.join(OUTPUT_DIR, "1_out.jpg"))
     return os.path.join(OUTPUT_DIR, "1_out.jpg")
 
 input_image = gr.inputs.Image(type="pil", label="Input")

test.py

@@ -22,12 +22,14 @@ model = model.to(device)
 base = os.path.splitext(os.path.basename(img_path))[0]
 
 # read image
-print(img_path)
+print(img_path);
 img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)
 img = torch.from_numpy(img[np.newaxis, :, :]).float()
 img_LR = img.unsqueeze(0)
 img_LR = img_LR.to(device)
 
+print('Start upscaling...')
 output = model(img_LR).squeeze(dim=0).float().cpu().clamp_(0, 1).numpy()
 output = np.transpose(output, (1, 2, 0))
+print('Finished upscaling, saving image.')
 cv2.imwrite(output_dir, output, [int(cv2.IMWRITE_JPEG_QUALITY), 90])