Update app.py

app.py

@@ -25,7 +25,7 @@ def process_image_and_prompt(input_image, prompt):
     depth_estimator = pipeline("depth-estimation")
 
     # Get depth map
-    depth_map = get_depth_map(input_image, depth_estimator).unsqueeze(0).half().to("cuda")
+    depth_map = get_depth_map(input_image, depth_estimator).unsqueeze(0).half().to("cpu")
     
     # Load the ControlNet model and the StableDiffusionControlNetImg2ImgPipeline
     controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-normal", torch_dtype=torch.float16, use_safetensors=True)
@@ -34,7 +34,7 @@ def process_image_and_prompt(input_image, prompt):
         controlnet=controlnet,
         torch_dtype=torch.float16,
         use_safetensors=True
-    ).to("cuda")
+    ).to("cpu")
     pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
     pipe.enable_model_cpu_offload()