Update app.py

app.py

@@ -60,7 +60,8 @@ def align_embeddings(prompt_embeds, negative_prompt_embeds):
            torch.nn.functional.pad(negative_prompt_embeds, (0, 0, 0, max_length - negative_prompt_embeds.shape[1]))
 
 pipe_default = get_lora_sd_pipeline(lora_dir='./lora_man_animestyle', base_model_name_or_path=model_default, dtype=torch_dtype).to(device)
-pipe_controlnet = StableDiffusionControlNetPipeline.from_pretrained(
+#pipe_controlnet = StableDiffusionControlNetPipeline.from_pretrained(
+pipe_controlnet = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(    
     model_default,
     controlnet=controlnet,
     torch_dtype=torch_dtype