NickKolok commited on
Commit
4062990
·
verified ·
1 Parent(s): 8388c54

Debug output

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -10,8 +10,11 @@ os.system('wget https://github.com/JingyunLiang/SwinIR/releases/download/v0.0/00
10
  def inference(img):
11
  os.system('mkdir -p test')
12
  img.save("test/1.png", "PNG")
 
13
  os.system('python main_test_swinir.py --task real_sr --model_path experiments/pretrained_models/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.pth --folder_lq test --scale 4')
 
14
  os.system("optipng -o5 results/swinir_real_sr_x4/1_SwinIR.png")
 
15
  return 'results/swinir_real_sr_x4/1_SwinIR.png'
16
 
17
  title = "SwinIR"
 
10
  def inference(img):
11
  os.system('mkdir -p test')
12
  img.save("test/1.png", "PNG")
13
+ print("Image saved!")
14
  os.system('python main_test_swinir.py --task real_sr --model_path experiments/pretrained_models/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.pth --folder_lq test --scale 4')
15
+ print("Upscale performed!")
16
  os.system("optipng -o5 results/swinir_real_sr_x4/1_SwinIR.png")
17
+ print("optipng performed!")
18
  return 'results/swinir_real_sr_x4/1_SwinIR.png'
19
 
20
  title = "SwinIR"