Spaces:

esun-choi
/

INVOHIDE_inisw8

Build error

esun-choi commited on Dec 19, 2023

Commit

6d6ace0

1 Parent(s): 088dab0

Update sr/sr.py

Files changed (1) hide show

sr/sr.py CHANGED Viewed

@@ -3,8 +3,8 @@ from PIL import Image
 import numpy as np
 from .esrgan.RealESRGAN import RealESRGAN
 def sr(img):
-   # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-    device=torch.device("cuda")
     model = RealESRGAN(device, scale=2)
     model.load_weights('weights/RealESRGAN_x2.pth', download=True)

 import numpy as np
 from .esrgan.RealESRGAN import RealESRGAN
 def sr(img):
+    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    # device=torch.device("cuda")
     model = RealESRGAN(device, scale=2)
     model.load_weights('weights/RealESRGAN_x2.pth', download=True)