switch between gpu and cpu

app.py

@@ -17,13 +17,14 @@ pn.state.template.param.update(
 )
 
 model_id = "timbrooks/instruct-pix2pix"
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 if 'pipe' in pn.state.cache:
     pipe = pn.state.cache['pipe']
 else:
     pipe = pn.state.cache['pipe'] = StableDiffusionInstructPix2PixPipeline.from_pretrained(
         model_id, torch_dtype=torch.float16
-    ).to("cuda")
+    ).to(device)
     pipe.enable_xformers_memory_efficient_attention()
     pipe.unet.to(memory_format=torch.channels_last)
 
@@ -38,6 +39,9 @@ def normalize_image(value, width):
     return image.resize((width, height), PIL.Image.LANCZOS)
 
 def new_image(prompt, image, img_guidance, guidance, steps, width=600):
+    """
+    create a new image from the StableDiffusionInstructPix2PixPipeline model
+    """
     edit = pipe(
         prompt,
         image=image,