Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -17,6 +17,9 @@ from preprocess.humanparsing.run_parsing import Parsing
 from ootd.inference_ootd_hd import OOTDiffusionHD
 from ootd.inference_ootd_dc import OOTDiffusionDC
 
+from huggingface_hub import snapshot_download
+large_files = ["checkpoints/humanparsing/parsing_atr.onnx", "checkpoints/humanparsing/parsing_lip.onnx", "checkpoints/ootd/ootd_dc/checkpoint-36000/unet_garm/diffusion_pytorch_model.safetensors", "checkpoints/ootd/ootd_dc/checkpoint-36000/unet_vton/diffusion_pytorch_model.safetensors", "checkpoints/ootd/ootd_hd/checkpoint-36000/unet_garm/diffusion_pytorch_model.safetensors", "checkpoints/ootd/ootd_hd/checkpoint-36000/unet_vton/diffusion_pytorch_model.safetensors", "checkpoints/ootd/text_encoder/pytorch_model.bin", "checkpoints/ootd/vae/diffusion_pytorch_model.bin"]
+snapshot_download(repo_id="John6666/OOTDiffusion-storage", repo_type="model", allow_patterns=large_files, local_dir=".")
 
 openpose_model_hd = OpenPose(0)
 parsing_model_hd = Parsing(0)