BrushEdit

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">

Video

Watch the introduction video in our project page or YouTube.

Code

Please check our GitHub repository for code.

Model

Download the model checkpoint using huggingface_hub (Version 0.1 as example):

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.