Update app.py

app.py

@@ -5,10 +5,11 @@ import numpy as np
 from diffusers import StableDiffusionDepth2ImgPipeline
 from pathlib import Path
 
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 dept2img = StableDiffusionDepth2ImgPipeline.from_pretrained(
     "stabilityai/stable-diffusion-2-depth",
     torch_dtype=torch.float16,
-).to("cuda")
+).to(device)
 
 
 def pad_image(input_image):