Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -430,7 +430,7 @@ CHECKPOINTS = ESRGANUpscalerCheckpoints(
|
|
430 |
device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
431 |
DTYPE = dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
|
432 |
|
433 |
-
enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=device, dtype=DTYPE)
|
434 |
|
435 |
# logging
|
436 |
|
|
|
430 |
device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
431 |
DTYPE = dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
|
432 |
|
433 |
+
enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=device, dtype=DTYPE)
|
434 |
|
435 |
# logging
|
436 |
|