Update app.py

app.py

@@ -93,10 +93,11 @@ def infer(
                                                                  controlnet=controlnet,
                                                                  torch_dtype=torch_dtype, 
                                                                  safety_checker=None).to(device)
+        print(type(controlnet_image))
+        controlnet_image = load_image(controlnet_image).convert('RGB')
         params['control_image'] = controlnet_image
         params['controlnet_conditioning_scale'] = controlnet_strength
-        # controlnet_image = load_image(controlnet_image).convert('RGB')
-        # print(type(controlnet_image))
+        print(type(controlnet_image))
     else:
         pipe = StableDiffusionPipeline.from_pretrained(model_id, 
                                                        torch_dtype=torch_dtype, 
@@ -106,8 +107,8 @@ def infer(
     if ip_adapter_checkbox:
         pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")
         pipe.set_ip_adapter_scale(ip_adapter_scale)
+        ip_adapter_image = load_image(ip_adapter_image).convert('RGB') 
         params['ip_adapter_image'] = ip_adapter_image
-        # ip_adapter_image = load_image(ip_adapter_image).convert('RGB') 
     
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir)
     pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir)