Update app.py

app.py

@@ -4,13 +4,15 @@ import numpy as np
 import gradio as gr
 from pytorch_lightning import seed_everything
 from annotator.util import resize_image, HWC3
-from diffusers import StableDiffusionControlNetPipeline
+from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
+
+# Load the controlnet model
+controlnet = ControlNetModel.from_pretrained("path/to/controlnet/model")
 
 # Load the pipeline
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
-    controlnet="path/to/controlnet/model",
-    image_encoder="path/to/image_encoder/model"
+    controlnet=controlnet
 ).to("cuda")
 
 def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode, strength, scale, seed, eta, low_threshold, high_threshold):