Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -426,7 +426,7 @@ CHECKPOINTS = ESRGANUpscalerCheckpoints(
|
|
426 |
|
427 |
device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
428 |
DTYPE = dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
|
429 |
-
enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=
|
430 |
|
431 |
# logging
|
432 |
|
@@ -662,7 +662,6 @@ def add_song_cover_text(img,artist,song,h,w):
|
|
662 |
|
663 |
return img
|
664 |
|
665 |
-
spaces.GPU(300)
|
666 |
def upscale_image(img,factor=2):
|
667 |
|
668 |
global working
|
|
|
426 |
|
427 |
device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
428 |
DTYPE = dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
|
429 |
+
enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device="cpu", dtype=DTYPE)
|
430 |
|
431 |
# logging
|
432 |
|
|
|
662 |
|
663 |
return img
|
664 |
|
|
|
665 |
def upscale_image(img,factor=2):
|
666 |
|
667 |
global working
|