Update app.py

app.py

@@ -39,7 +39,6 @@ def infer(
     ip_adapter_image=None,
     progress=gr.Progress(track_tqdm=True),    
 ):  
-    print(controlnet_checkbox, ip_adapter_checkbox)
     generator = torch.Generator(device).manual_seed(seed)
 
     ckpt_dir='./model_output'
@@ -80,10 +79,10 @@ def infer(
                 cache_dir="./models_cache",
                 torch_dtype=torch_dtype
             )
-        pipe = StableDiffusionPipeline.from_pretrained(model_id, 
-                                                       controlnet=controlnet,
-                                                       torch_dtype=torch_dtype, 
-                                                       safety_checker=None).to(device)
+        pipe = StableDiffusionControlNetPipeline.from_pretrained(model_id, 
+                                                                 controlnet=controlnet,
+                                                                 torch_dtype=torch_dtype, 
+                                                                 safety_checker=None).to(device)
     else:
         pipe = StableDiffusionPipeline.from_pretrained(model_id, 
                                                        torch_dtype=torch_dtype,