Spaces:

52Hz
/

SRMNet_real_world_denoising

Runtime error

52Hz commited on Jan 14, 2022

Commit

3b59ff7

1 Parent(s): be936b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ def inference(img):
     img.save("test/1.png", "PNG")
     os.system(
         'python main_test_SRMNet.py --input_dir test --weights experiments/pretrained_models/AWGN_denoising_SRMNet.pth')
-    return 'result/out.png'
 title = "Selective Residual M-Net (SRMNet)"

     img.save("test/1.png", "PNG")
     os.system(
         'python main_test_SRMNet.py --input_dir test --weights experiments/pretrained_models/AWGN_denoising_SRMNet.pth')
+    return 'result/1.png'
 title = "Selective Residual M-Net (SRMNet)"