Fix style

app.py

@@ -5,7 +5,6 @@ from PIL import Image
 import qrcode
 from qrcode.constants import ERROR_CORRECT_H
 
-
 # ========= device/dtype =========
 device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
 dtype = torch.float16 if device != "cpu" else torch.float32
@@ -44,21 +43,20 @@ def make_qr(url: str = "http://www.mybirdfire.com", size: int = 512, border: int
     img = qr.make_image(fill_color="black", back_color="white").convert("RGB")
     return img.resize((size, size), resample=Image.NEAREST)
 
-# ========= ControlNet Stylizer (SD1.5 + sd15-canny) =========
-# --- SDXL dual ControlNet stylizer (canny + tile) ---
+# ========= SDXL dual ControlNet stylizer (canny + softedge) =========
 from diffusers import StableDiffusionXLControlNetPipeline, ControlNetModel
 from diffusers.schedulers.scheduling_euler_discrete import EulerDiscreteScheduler
 from controlnet_aux import CannyDetector
 
-SDXL_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"   # swap to your SDXL anime checkpoint if you have one
+SDXL_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"   # swap to your SDXL anime model if desired
 CN_CANNY   = "diffusers/controlnet-canny-sdxl-1.0"
-CN_TILE    = "diffusers/controlnet-tile-sdxl-1.0"
+CN_SOFT    = "diffusers/controlnet-softedge-sdxl-1.0"     # <-- replaces non-existent tile SDXL
 
 _sdxl = {"pipe": None}
 def _load_sdxl_dual():
     if _sdxl["pipe"] is None:
         cn1 = ControlNetModel.from_pretrained(CN_CANNY, torch_dtype=dtype)
-        cn2 = ControlNetModel.from_pretrained(CN_TILE,   torch_dtype=dtype)
+        cn2 = ControlNetModel.from_pretrained(CN_SOFT,  torch_dtype=dtype)
         pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
             SDXL_MODEL, controlnet=[cn1, cn2], torch_dtype=dtype, safety_checker=None
         ).to(device)
@@ -80,7 +78,7 @@ def stylize_qr_sdxl(prompt: str, steps: int=28, guidance: float=7.0, seed: int=1
 
     gen = torch.Generator(device=device).manual_seed(int(seed)) if int(seed)!=0 else None
 
-    # Control weights + schedule (canny, tile)
+    # Control weights + schedule (canny, softedge)
     cn_scales  = [1.1, 0.6]
     cn_start   = [0.25, 0.00]
     cn_end     = [0.95, 1.00]
@@ -89,18 +87,14 @@ def stylize_qr_sdxl(prompt: str, steps: int=28, guidance: float=7.0, seed: int=1
         img = pipe(
             prompt=prompt,
             negative_prompt=NEG,
-            image=[edges, qr],                               # canny first, tile second
-            controlnet_conditioning_scale=cn_scales,         # strengths
-            control_guidance_start=cn_start,                 # when they start
-            control_guidance_end=cn_end,                     # when they end
+            image=[edges, qr],                               # canny first, softedge second
+            controlnet_conditioning_scale=cn_scales,
+            control_guidance_start=cn_start,
+            control_guidance_end=cn_end,
             num_inference_steps=int(steps),
             guidance_scale=float(guidance),
             generator=gen
         ).images[0]
-
-        # optional: re‑overlay razor‑sharp finder squares to boost scanning
-        # (uncomment if scans are borderline)
-        # img = overlay_finders(img, qr)
         return img
 
     if device in ("cuda", "mps"):
@@ -108,7 +102,6 @@ def stylize_qr_sdxl(prompt: str, steps: int=28, guidance: float=7.0, seed: int=1
             return run()
     return run()
 
-
 # ========= UI =========
 with gr.Blocks() as demo:
     gr.Markdown("## Stable Diffusion + QR Code + ControlNet")
@@ -129,7 +122,7 @@ with gr.Blocks() as demo:
         out_qr = gr.Image(label="QR Code", type="pil")
         gr.Button("Generate QR").click(make_qr, [url, size, quiet], out_qr)
         
-    with gr.Tab("QR Stylizer (SDXL dual ControlNet)"):
+    with gr.Tab("QR Stylizer (SDXL canny + softedge)"):
         p = gr.Textbox(label="Prompt", value="Sky, Moon, Bird, Blue, In the dark, Goddess, Sweet, Beautiful, Fantasy, Art, Anime")
         st = gr.Slider(20, 40, 28, step=1, label="Steps")
         cfg = gr.Slider(4.5, 9.0, 7.0, step=0.1, label="CFG")
@@ -139,6 +132,5 @@ with gr.Blocks() as demo:
         out = gr.Image(label="Stylized QR (SDXL)")
         gr.Button("Stylize").click(stylize_qr_sdxl, [p, st, cfg, sd, cl, ch], out)
 
-
 if __name__ == "__main__":
     demo.launch()

requirements.txt

@@ -5,4 +5,4 @@ accelerate
 safetensors
 gradio
 qrcode[pil]
-controlnet-aux
+controlnet-aux