Update README.md

README.md

@@ -10,10 +10,10 @@ base_model:
 
 # BlobCtrl
 
-<a href=''><img src="https://img.shields.io/static/v1?label=Arxiv Preprint&message=BlobCtrl&color=red&logo=arxiv"></a>
+<a href='http://arxiv.org/abs/2503.13434'><img src="https://img.shields.io/static/v1?label=Arxiv Preprint&message=BlobCtrl&color=red&logo=arxiv"></a>
 <a href='https://liyaowei-stu.github.io/project/BlobCtrl/'><img src='https://img.shields.io/badge/Project-Page-green'></a>
-<a href=''><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Demo-blue'></a> 
-<a href=''><img src='https://img.shields.io/badge/Github-Repo-blue'></a>
+<a href='https://huggingface.co/spaces/Yw22/BlobCtrl'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Demo-blue'></a> 
+<a href='https://github.com/TencentARC/BlobCtrl'><img src='https://img.shields.io/badge/Github-Repo-blue'></a>
 
 ### Overview 
 
@@ -23,11 +23,11 @@ Main Features: 🦉Element-level Add/Remove/Move/Replace/Enlarge/Shrink.
 
 
 ### Video
-[Watch the introduction video](https://liyaowei-stu.github.io/project/BlobCtrl) in our project page or [YouTube]().
+[Watch the introduction video](https://liyaowei-stu.github.io/project/BlobCtrl) in our project page or [YouTube](https://youtu.be/rdR4QRR-mbE).
 
 ### Code
 
-Please check our [GitHub repository]() for code.
+Please check our [GitHub repository](https://github.com/TencentARC/BlobCtrl) for code.
 
 
 
@@ -39,14 +39,15 @@ Download the model checkpoint using `huggingface_hub` (Version 0.1 as example):
 import os
 from huggingface_hub import snapshot_download
 
-# download hf models
+# download blobctrl models
 BlobCtrl_path = "examples/blobctrl/models"
-if not os.path.exists(BlobCtrl_path):
-    BrushEdit_path = snapshot_download(
+if not (os.path.exists(f"{BlobCtrl_path}/blobnet") and os.path.exists(f"{BlobCtrl_path}/unet_lora")):
+    BlobCtrl_path = snapshot_download(
         repo_id="Yw22/BlobCtrl",
-        local_dir=BrushEdit_path,
+        local_dir=BlobCtrl_path,
         token=os.getenv("HF_TOKEN"),
     )
+print(f"BlobCtrl checkpoints downloaded to {BlobCtrl_path}")    )
 ```
  
 The downloaded BlobCtrl checkpoint file (blobnet&unet_lora) can be found at `BlobCtrl_path`.
@@ -54,4 +55,4 @@ The downloaded BlobCtrl checkpoint file (blobnet&unet_lora) can be found at `Blo
 
 
 ### Demo
-You can [try the demo here]().
+You can [try the demo here](https://huggingface.co/spaces/Yw22/BlobCtrl).