BrushEdit

Overview

BrushEdit is an advanced, unified AI agent for image inpainting and editing. Main Elements: 🛠️ Fully automated / 🤠 Interactive editing.

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.