Yaron Koresh commited on
Commit
8172c90
·
verified ·
1 Parent(s): ba4ed60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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=DEVICE, dtype=DTYPE)
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