52Hz commited on
Commit
a530eb8
1 Parent(s): d8e88ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def inference(img):
16
  img.save("test/1.png", "PNG")
17
  os.system(
18
  'python main_test_SRMNet.py --weights experiments/pretrained_models/AWGN_denoising_SRMNet.pth')
19
- return 'result/1.png'
20
 
21
 
22
  title = "Selective Residual M-Net (SRMNet)"
 
16
  img.save("test/1.png", "PNG")
17
  os.system(
18
  'python main_test_SRMNet.py --weights experiments/pretrained_models/AWGN_denoising_SRMNet.pth')
19
+ return 'result/out.png'
20
 
21
 
22
  title = "Selective Residual M-Net (SRMNet)"