Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -616,8 +616,8 @@ class ImageRestorer:
|
|
| 616 |
tensor = (tensor * 0.5 + 0.5).clamp(0, 1)
|
| 617 |
arr = (tensor.permute(1, 2, 0).numpy() * 255).astype(np.uint8)
|
| 618 |
return Image.fromarray(arr)
|
| 619 |
-
|
| 620 |
-
|
| 621 |
def restore(
|
| 622 |
self,
|
| 623 |
image: Image.Image,
|
|
|
|
| 616 |
tensor = (tensor * 0.5 + 0.5).clamp(0, 1)
|
| 617 |
arr = (tensor.permute(1, 2, 0).numpy() * 255).astype(np.uint8)
|
| 618 |
return Image.fromarray(arr)
|
| 619 |
+
|
| 620 |
+
@torch.no_grad()
|
| 621 |
def restore(
|
| 622 |
self,
|
| 623 |
image: Image.Image,
|