set up device

app.py

@@ -15,11 +15,21 @@ pn.state.template.main_max_width = "690px"
 pn.state.template.accent_base_color = "#F08080"
 pn.state.template.header_background = "#F08080"
 
+# Set up device 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
 # Model
 model_id = "timbrooks/instruct-pix2pix"
-pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
-    model_id, torch_dtype=torch.float16
-).to("cuda")
+
+if devide == "cuda":
+    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
+        model_id, torch_dtype=torch.float16
+    )
+else:
+    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
+        model_id
+    )
+pipe = pipe.to(device)
 pipe.enable_xformers_memory_efficient_attention()
 pipe.unet.to(memory_format=torch.channels_last)