Update app.py

app.py

@@ -3,7 +3,8 @@ import torch
 import numpy as np
 import cv2
 from diffusers import StableDiffusionPipeline
-from model import UNet2DConditionModelEx, StableDiffusionControlLoraV3Pipeline
+from model import UNet2DConditionModelEx
+from pipeline import StableDiffusionControlLoraV3Pipeline 
 from PIL import Image
 import os
 from huggingface_hub import login
@@ -22,7 +23,7 @@ unet = UNet2DConditionModelEx.from_pretrained(
     torch_dtype=dtype
 )
 
-# Add conditioning
+# Add conditioning with ow-gbi-control-lora
 unet = unet.add_extra_conditions("ow-gbi-control-lora")
 
 # Create the pipeline with custom UNet
@@ -58,7 +59,8 @@ def generate_image(input_image, prompt, negative_prompt, guidance_scale, steps,
             negative_prompt=negative_prompt,
             num_inference_steps=steps,
             guidance_scale=guidance_scale,
-            image=canny_image
+            image=canny_image,
+            extra_condition_scale=1.0
         ).images[0]
     
     return canny_image, image