Spaces:

deepinv
/

denoising

Sleeping

Julián Tachella commited on Feb 23, 2024

Commit

a4dc15b

1 Parent(s): fd548c6

test

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def image_mod(image, noise_level, denoiser):
     else:
         raise ValueError("Invalid denoiser")
     noisy = image + torch.randn_like(image) * noise_level
-    estimated = denoiser(image, noise_level)
     return torch_to_pil(noisy), torch_to_pil(estimated)

     else:
         raise ValueError("Invalid denoiser")
     noisy = image + torch.randn_like(image) * noise_level
+    estimated = denoiser(noisy, noise_level)
     return torch_to_pil(noisy), torch_to_pil(estimated)