change to cuda

app.py

@@ -15,7 +15,7 @@ from PIL import Image, ImageOps
 
 # Constants
 TARGET_SIZE = (512, 512)
-DEVICE = torch.device("cpu")
+DEVICE = torch.device("cuda")
 LPIPS_MODELS = ['alex', 'vgg', 'squeeze']
 MASK_SIZES = {"64x64": 64, "128x128": 128, "256x256": 256}
 DEFAULT_MASK_SIZE = "256x256"
@@ -250,4 +250,4 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     demo = create_gradio_interface()
-    demo.launch(allowed_paths=["static"])
+    demo.launch(allowed_paths=["static"])