File size: 1,806 Bytes
5e41ed7 f4622fc 5e41ed7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
---
license: other
language:
- en
pipeline_tag: image-to-image
library_name: diffusers
---
# BrushEdit
<a href="https://arxiv.org/abs/2412.10316"><img src="https://img.shields.io/static/v1?label=Arxiv Preprint&message=BrushEdit&color=red&logo=arxiv"></a>
<a href='https://liyaowei-stu.github.io/project/BrushEdit/'><img src='https://img.shields.io/badge/Project-Page-green'></a>
<a href='https://huggingface.co/spaces/TencentARC/BrushEdit'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Demo-blue'></a>
<a href='https://github.com/TencentARC/BrushEdit.git'><img src='https://img.shields.io/badge/Github-Repo-blue'></a>
### Overview
BrushEdit is an advanced, unified AI agent for image inpainting and editing.
Main Elements: 🛠️ Fully automated / 🤠 Interactive editing.
[<img src="https://github.com/user-attachments/assets/c8a23ee0-712b-4f68-a806-55008b5b381d" width="600">](https://liyaowei-stu.github.io/project/BrushEdit)
### Video
[Watch the introduction video](https://liyaowei-stu.github.io/project/BrushEdit/) in our project page or [YouTube](https://www.youtube.com/watch?v=nDB7un9Rbdk).
### Code
Please check our [GitHub repository](https://github.com/TencentARC/BrushEdit.git) for code.
### Model
Download the model checkpoint using `huggingface_hub` (Version 0.1 as example):
```python
import os
from huggingface_hub import snapshot_download
# download hf models
BrushEdit_path = "models/"
if not os.path.exists(BrushEdit_path):
BrushEdit_path = snapshot_download(
repo_id="TencentARC/BrushEdit",
local_dir=BrushEdit_path,
token=os.getenv("HF_TOKEN"),
)
```
The downloaded checkpoint file can be found at `BrushEdit_path`.
### Demo
You can [try the demo here](https://huggingface.co/spaces/TencentARC/BrushEdit).
|