Update app.py

app.py

@@ -11,7 +11,10 @@ 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"
+#model_path = "Skywork/Skywork-UniPic-1.5B"
+from huggingface_hub import hf_hub_download
+
+model_path = hf_hub_download(repo_id="Skywork/Skywork-UniPic-1.5B", filename="pytorch_model.bin")
 ckpt_name = "UniPic"
 
 inferencer = Inferencer(