📔 Table of Contents

• 📔 Table of Contents
  • 🌟 About the Project
    • 📷 Screenshots
    • 👾 Tech Stack
  • 🧰 Getting Started
    • ‼️ Prerequisites
    • 🏃 Run Locally
  • 👀 Usage
  • 👋 Contributing
    • 📜 Code of Conduct
  • ⚠️ License
  • 🤝 Contact

🌟 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

• React.js
• TailwindCSS
• FastAPI

🧰 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 - [email protected]

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