Update app.py

app.py

@@ -67,10 +67,13 @@ def infer(
     pipe = None
     if controlnet_enabled and control_image:
         controlnet_model = ControlNetModel.from_pretrained(CONTROLNET_MODES.get(control_mode))
-        if model_id == "SD1.5 + lora Unet TextEncoder" or model_id == "SD1.5 + lora Unet":
+        if model_id == "SD1.5 + lora Unet TextEncoder": 
             pipe=StableDiffusionControlNetPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5",controlnet=controlnet_model)
             pipe.unet = PeftModel.from_pretrained(pipe.unet, "um235/vCat_v2", subfolder="unet")
             pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, "um235/vCat_v2", subfolder="text_encoder")
+        elif model_id == "SD1.5 + lora Unet TextEncoder" or model_id == "SD1.5 + lora Unet":
+            pipe=StableDiffusionControlNetPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5",controlnet=controlnet_model)
+            pipe.unet = PeftModel.from_pretrained(pipe.unet, "um235/cartoon_cat_stickers")
         else:
             pipe=StableDiffusionControlNetPipeline.from_pretrained(model_id, controlnet=controlnet_model)
     else: