Hugging Face's logo

Spaces:
gonzmart
/
recursive_inpainting
Running

App Files Community
gonzmart commited on
Commit
e9121e4
•
1 Parent(s): de516af

change to cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from PIL import Image, ImageOps
15
 
16
  # Constants
17
  TARGET_SIZE = (512, 512)
18
- DEVICE = torch.device("cpu")
19
  LPIPS_MODELS = ['alex', 'vgg', 'squeeze']
20
  MASK_SIZES = {"64x64": 64, "128x128": 128, "256x256": 256}
21
  DEFAULT_MASK_SIZE = "256x256"
 
15
 
16
  # Constants
17
  TARGET_SIZE = (512, 512)
18
+ DEVICE = torch.device("cuda")
19
  LPIPS_MODELS = ['alex', 'vgg', 'squeeze']
20
  MASK_SIZES = {"64x64": 64, "128x128": 128, "256x256": 256}
21
  DEFAULT_MASK_SIZE = "256x256"