52Hz commited on
Commit
f573f7f
1 Parent(s): 8bdc2fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def inference(img):
15
  img = img.resize((basewidth, hsize), Image.ANTIALIAS)
16
  img.save("test/1.png", "PNG")
17
  os.system(
18
- 'python main_test_SRMNet.py --input_dir test --weights experiments/pretrained_models/AWGN_denoising_SRMNet.pth')
19
  return 'result/1.png'
20
 
21
 
 
15
  img = img.resize((basewidth, hsize), Image.ANTIALIAS)
16
  img.save("test/1.png", "PNG")
17
  os.system(
18
+ 'python main_test_SRMNet.py --input_dir test --weights experiments/pretrained_models/real_denoising_SRMNet.pth')
19
  return 'result/1.png'
20
 
21