Beta-VAE Latent Space Editing

Demonstration of latent space editing of face attributes using a Beta-VAE trained on CelebA.
Model card: Hugging Face
GitHub: Repository

If you upload a custom image, make sure it is square and the face is centered with minimal background, since the image will be center-cropped before being processed.

Upload Image

Or select sample images

Sample Faces

Original (Left) & Reconstruction (Right)

Image

Image

Edited Image

Image

Edit Dimensions

Feature to Edit

Edit Strength

-3 3

Positive values increase the attribute, while negative values reduce it.