Update app.py

app.py

@@ -256,7 +256,7 @@ def predict_mask(segmentmodel, sam, image, TEXT_PROMPT, segmentType, confidence
     return masks
 
 def build_model_sd(pretrained_model, controlnet_path, face_adapter, device, prompts, antelopev2_path, width, height, style_lora):
-    controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16)
+    controlnet = ControlNetModel.from_pretrained(controlnet_path, torch_dtype=torch.float16).to(device)
     pipe = InstantidMultiConceptPipeline.from_pretrained(
         pretrained_model, controlnet=controlnet, torch_dtype=torch.float16).to(device)