Update app.py

app.py

@@ -12,9 +12,9 @@ from accelerate.utils import set_seed
 # model_path = "/mnt/data_vlm/chris/Harmon/work_dirs/sft_stage4_mar_huge_lr_7e-6_edit_mainly_new_mar_few_steps_gpt_only_and_1024/500iters_hf/pytorch_model.bin"
 # ckpt_name = "/".join(model_path.split("/")[-3:])
 #model_path = "Skywork/Skywork-UniPic-1.5B"
-from huggingface_hub import hf_hub_download
+from huggingface_hub import snapshot_download
 
-model_path = hf_hub_download(repo_id="Skywork/Skywork-UniPic-1.5B")
+model_path = snapshot_download(repo_id="Skywork/Skywork-UniPic-1.5B")
 ckpt_name = "UniPic"
 
 inferencer = Inferencer(