Update app.py

app.py

@@ -29,16 +29,17 @@ qrcode_generator = qrcode.QRCode(
 )
 
 controlnet = ControlNetModel.from_pretrained(
-    "monster-labs/control_v1p_sd15_qrcode_monster", torch_dtype=torch.float16
+    "DionTimmer/controlnet_qrcode-control_v1p_sd15", torch_dtype=torch.float16
 ).to("cuda")
 
-pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
-    "rupeshs/LCM-runwayml-stable-diffusion-v1-5",
-    controlnet=controlnet,
-    safety_checker=None,
-    torch_dtype=torch.float16,
+pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    # "runwayml/stable-diffusion-v1-5",
+    "digiplay/GhostMixV1.2VAE",
+    controlnet = controlnet,
+    torch_dtype = torch.float16,
+    safety_checker =None,
 ).to("cuda")
-# pipe.enable_xformers_memory_efficient_attention()
+#pipe.enable_xformers_memory_efficient_attention()
 
 
 def resize_for_condition_image(input_image: Image.Image, resolution: int):