Update app.py

app.py

@@ -9,8 +9,8 @@ import torchvision.transforms.functional as F
 from diffusers import ControlNetModel, StableDiffusionControlNetPipeline
 import gradio as gr
 
-device = "cpu"   # Linux & Windows
-weight_type = torch.float32  # torch.float16 works as well, but pictures seem to be a bit worse
+device = "cuda"
+weight_type = torch.float16
 
 controlnet = ControlNetModel.from_pretrained(
     "IDKiro/sdxs-512-dreamshaper-sketch", torch_dtype=weight_type
@@ -88,7 +88,7 @@ def run(
     prompt_template, 
     style_name, 
     controlnet_conditioning_scale,
-    device_type="CPU",
+    device_type="GPU",
     param_dtype='torch.float16',
 ):
     if device_type == "CPU":