Julián Tachella commited on
Commit ·
a4dc15b
1
Parent(s): fd548c6
test
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ def image_mod(image, noise_level, denoiser):
|
|
| 46 |
else:
|
| 47 |
raise ValueError("Invalid denoiser")
|
| 48 |
noisy = image + torch.randn_like(image) * noise_level
|
| 49 |
-
estimated = denoiser(
|
| 50 |
return torch_to_pil(noisy), torch_to_pil(estimated)
|
| 51 |
|
| 52 |
|
|
|
|
| 46 |
else:
|
| 47 |
raise ValueError("Invalid denoiser")
|
| 48 |
noisy = image + torch.randn_like(image) * noise_level
|
| 49 |
+
estimated = denoiser(noisy, noise_level)
|
| 50 |
return torch_to_pil(noisy), torch_to_pil(estimated)
|
| 51 |
|
| 52 |
|