kabir12345 / godesigner

GoDesigner is a B2B tool designed for interior designers. It allows users to upload photos of homes and reimagine the house using generative AI. The web app also provides product recommendations that are present in the generated photo using the Segment Anything model and Google Shopping API.

Sure, here's the detailed and technical README.md for your project:

GoDesigner

Table of Contents

1. Overview
2. Technologies Used
3. Installation
4. Usage
5. Contributing
6. Known Issues
7. Future Improvements
8. Contact Information
9. License

Overview

GoDesigner is a B2B tool designed for interior designers. It allows users to upload photos of homes and reimagine the house using generative AI. The web app also provides product recommendations that are present in the generated photo using the Segment Anything model and Google Shopping API.

Technologies Used

Programming Language

The primary language used for this project is Python, known for its readability and efficiency in data analysis and AI model development.

AI Model

The core of this project is a fine-tuned Stable Diffusion model. This model is a type of generative model that is specifically tailored towards images and prompts related to the images. Here are some key details about the model:

• Model Type: stabilityai/stable-diffusion
• Training: The model has been fine-tuned on a custom dataset.
• Dataset: The dataset used for training consists of images and related prompts. The dataset has a population size of 10798.

Product Recommendation

For product recommendation, the project utilizes the Segment Anything model and Google Shopping API. The Segment Anything model identifies products in the generated images, and the Google Shopping API is used to find similar products online.

Containerization

Docker is used for containerizing the application. Containerization allows the application to run in an isolated environment, removing the hassle of dependency management and ensuring that the application runs the same way in every environment.

Web Application

The user interface of the application is a web app, allowing users to interact with the AI model conveniently through their web browser.

Installation

To install and run the project, follow these steps:

1. Clone the repository:

git clone https://github.com/kabir12345/godesigner.git

2. Navigate to the project directory:

cd godesigner

3. Install the required dependencies:

pip install -r requirements.txt

4. Build the Docker image:

docker build -t godesigner .

5. Run the Docker container:

docker run -p 5000:5000 godesigner

Usage

Once the application is running, navigate to localhost:5000 in your web browser. From there, you can upload a photo of a home and the application will generate a reimagined version of the home using generative AI.

Contributing

Contributions are welcome! Please feel free to submit a Pull Request. For major changes, please open an issue first to discuss what you would like to change.

License

This project is licensed under the Apache-2.0 License - see the LICENSE file for details.