Professional Headshot with single image

Professional looking images with just one click!!

View Demo · Documentation · Report Bug · Request Feature

📔 Table of Contents

📔 Table of Contents

🌟 About the Project

It generates a professional headshot with just one image of the person. One click and get professional looking image. It uses Stable Diffusion.

📷 Screenshots

👾 Tech Stack

🧰 Getting Started

‼️ Prerequisites

Python3
npm

🏃 Run Locally

Clone the project

  git clone https://github.com/karanjakhar/Professional-Headshot-with-single-image.git

Go to the project directory

  cd Professional-Headshot-with-single-image

Install dependencies

  cd Professional-Headshot-with-single-image/frontend
  npm install

  cd Professional-Headshot-with-single-image/backend
  pip install -r requirements.txt

Run Backend and Frontend

  cd Professional-Headshot-with-single-image/frontend
  npm start

  cd Professional-Headshot-with-single-image/backend
  python3 -m app.main

Model Weights

Download model weight files from release and place them in weights/ folder.

Run Backend using Docker

cd Professional-Headshot-with-single-image/backend
docker build -t proshot:latest .
docker run --gpus all -p 8000:80 proshot:latest

👀 Usage

Project is in two part frontend (ReactJS) and backend (Python, FastAPI). To run it locally you need to run both.

👋 Contributing

Contributions are always welcome!

See contributing.md for ways to get started.

📜 Code of Conduct

Please read the Code of Conduct

⚠️ License

Distributed under Apache2.0. See LICENSE for more information.

🤝 Contact

Karan Jakhar - @karan-jakhar - karanjakhar49@gmail.com

Project Link: https://github.com/karanjakhar/Professional-Headshot-with-single-image