Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -10,12 +10,15 @@ from utils_ootd import get_mask_location
 PROJECT_ROOT = Path(__file__).absolute().parents[1].absolute()
 sys.path.insert(0, str(PROJECT_ROOT))
 
-import time
 from preprocess.openpose.run_openpose import OpenPose
 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 hf_hub_download
+
+hf_hub_download(repo_id="levihsu/OOTDiffusion", filename="checkpoints", local_dir="./")
+
 
 openpose_model_hd = OpenPose(0)
 parsing_model_hd = Parsing(0)