cache

app.py

@@ -17,9 +17,13 @@ pn.state.template.param.update(
 )
 
 model_id = "timbrooks/instruct-pix2pix"
-pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
-    model_id, torch_dtype=torch.float16
-).to("cuda")
+
+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")
 
 def normalize_image(value, width):
     """