Update app.py

app.py

@@ -40,7 +40,7 @@ if st.button("Generate"):
         farm_image = load_image_from_url(image_url)
         
         # Process image for depth map
-        depth_map = get_depth_map(farm_image, depth_estimator).unsqueeze(0).half().to("cuda")
+        depth_map = get_depth_map(farm_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)
@@ -49,7 +49,7 @@ if st.button("Generate"):
             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()