AI Portrait Generator (Demo)

I’m currently building an AI-powered system that transforms casual photos into studio-quality headshots. This demo is part of my ongoing exploration of image generation with Replicate and DreamBooth, combined with a Django backend and custom APIs.

Try it yourself (Demo)

Before (original photo)
After
Before (original photo)
After (img2img, trained model)
Before (original photo)
After (img2img, trained model)
AI generated txt2img, trained model
AI generated txt2img, trained model
AI generated txt2img, trained model
AI generated txt2img, trained model
AI generated txt2img, trained model
AI generated txt2img, trained model

Live image generation will be available soon.

How it works

  • Frontend: WordPress (this page)
  • Backend: Django + REST API
  • Model: Flux 1.1 Pro via Replicate
  • Image processing via Python & cloud-based inference
The user image is sent from WordPress to my Django backend, processed via the selected AI model on Replicate, and the result is returned as a URL preview.

Next steps

I’m currently training a custom DreamBooth model on RunPod to improve the realism and consistency of headshot generation. The goal is to offer high-quality, personalized headshots through an integrated service — directly from my own infrastructure.

Why I’m doing this

As a full-stack engineer with a focus on Python, AI integration, and modern frontend systems, I’m exploring how machine learning models can be used in creative, real-world applications. This project combines what I love: code, design, infrastructure, and future tech.

Nach oben scrollen