Update app.py

app.py

@@ -15,6 +15,12 @@ from diffusers import StableDiffusionXLPipeline
 import gradio as gr
 import gc
 
+
+from huggingface_hub import snapshot_download
+large_files = ["controlnet_depth/diffusion_pytorch_model.safetensors", "own_controlnet/diffusion_pytorch_model.safetensors]
+snapshot_download(repo_id="John6666/StableDesign-storage", repo_type="dataset", allow_patterns=large_files, local_dir=".")
+
+               
 device = "cuda"
 dtype = torch.float16