add caching

app.py

@@ -40,7 +40,7 @@ class Parts:
     UPPER = 1
     LOWER = 2
 
-@st.cache
+@st.cache(allow_output_mutation=True)
 def load_u2net():
     device = "cuda" if torch.cuda.is_available() else "cpu"
     checkpoint_path = hf_hub_download(repo_id="maiti/cloth-segmentation", filename="cloth_segm_u2net_latest.pth")
@@ -55,7 +55,7 @@ def change_bg_color(rgba_image, color):
     new_image.paste(rgba_image, (0, 0), rgba_image)
     return new_image.convert("RGB")
 
-@st.cache
+@st.cache(allow_output_mutation=True)
 def load_inpainting_pipeline():
     device = "cuda" if torch.cuda.is_available() else "cpu"
     inpainting_pipeline = StableDiffusionInpaintPipeline.from_pretrained(