Update app.py

app.py

@@ -54,7 +54,7 @@ controlnet_version = "coyo-500k"
 low_threshold = 100
 high_threshold = 200
 
-pipe, params = load_sb_pipe(controlnet_version)
+
 
 # pipe.enable_xformers_memory_efficient_attention()
 # pipe.enable_model_cpu_offload()
@@ -68,10 +68,13 @@ def pipe_inference(
     resolution=128,
     num_inference_steps=50,
     guidance_scale=7.5,
+    model="coyo-500k",
     seed=0,
     negative_prompt="",
     ):
-    print("Entered pipe...")
+    print("Loading pipe")
+    pipe, params = load_sb_pipe(model)
+        
     if not isinstance(image, np.ndarray):
         image = np.array(image) 
 
@@ -188,6 +191,11 @@ def create_demo(process, max_images=12, default_num_images=4):
                                                maximum=30.0,
                                                value=7.5,
                                                step=0.1)
+                    model = gr.Dropdown(choices=["coyo-500k", "bridge-2M"],
+                                        value="coyo-500k",
+                                        label="Model used for inference", 
+                                        info="Find every models at https://huggingface.co/Baptlem/baptlem-controlnet"
+                    ),
                     seed = gr.Slider(label='Seed',
                                      minimum=-1,
                                      maximum=2147483647,
@@ -213,6 +221,7 @@ def create_demo(process, max_images=12, default_num_images=4):
             #canny_high_threshold,
             num_steps,
             guidance_scale,
+            model,
             seed,
             n_prompt,
         ]