Update app.py

app.py

@@ -58,13 +58,19 @@ loaded_pipe = None
 def load_models_on_launch():
     global loaded_controlnet, loaded_pipe
     print("Loading models on launch...")
-    controlnet_path = snapshot_download(CONTROLNET_MODELS["QR Code Monster"])
+    
+    # Download the main repository
+    main_repo_path = snapshot_download("monster-labs/control_v1p_sd15_qrcode_monster")
+    
+    # Construct the path to the subfolder
+    controlnet_path = os.path.join(main_repo_path, "v2")
+    
     loaded_controlnet = ControlNetModel.from_pretrained(
         controlnet_path,
         torch_dtype=torch.float16
     ).to("cuda")
 
-    diffusion_path = snapshot_download(DIFFUSION_MODELS["GhostMix"])
+    diffusion_path = snapshot_download(DIFFUSION_MODELS["GhostMix", "Stable v1.5"])
     loaded_pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
         diffusion_path,
         controlnet=loaded_controlnet,