---
title: Stable Diffusion Image Generator
emoji: 🏢
colorFrom: blue
colorTo: purple
sdk: streamlit
sdk_version: 1.42.2
app_file: src/app.py
pinned: false
license: apache-2.0
short_description: Transform your ideas into artistic masterpieces 
---

# 🎨 AI Style Transfer Studio

Transform your ideas into artistic masterpieces using Stable Diffusion with custom style embeddings.

## 🚀 Features

- Multiple pre-trained style embeddings (Dhoni, Mickey Mouse, Balloon, Lion King, Rose Flower)
- Advanced color enhancement technology
- User-friendly Streamlit interface
- Real-time image generation
- Example gallery with style comparisons

## 🛠️ Local Setup

1. Clone the repository:
```bash
git clone https://github.com/yourusername/stable-diffusion-image-generator.git
cd stable-diffusion-image-generator
```

2. Create and activate a virtual environment (recommended):
```bash
python -m venv venv
# On Windows
venv\Scripts\activate
# On Unix or MacOS
source venv/bin/activate
```

3. Install dependencies:
```bash
pip install -r requirements.txt
```

4. Run the Streamlit app:
```bash
streamlit run src/app.py
```

The app will open in your default web browser at `http://localhost:8501`

## 🌐 Deploying to Hugging Face Spaces

1. Create a new Space on Hugging Face:
   - Go to https://huggingface.co/spaces
   - Click "Create new Space"
   - Choose "Streamlit" as the SDK
   - Set the Space name and visibility

2. Push your code to Hugging Face:
```bash
git add .
git commit -m "Initial commit"
git remote add space https://huggingface.co/spaces/yourusername/your-space-name
git push space main
```

3. The deployment will start automatically. Monitor the build logs on your Space's page.

## 🎯 Usage

1. Enter your creative prompt in the text area
2. Select a style from the available options
3. Click "Generate Artwork"
4. View both the original and color-enhanced versions of your creation

## 📝 Requirements

- Python 3.8+
- CUDA-capable GPU (recommended)
- 8GB+ RAM

## 🔑 Environment Variables

No additional environment variables are required for basic usage.

## 📄 License

This project is licensed under the Apache 2.0 License.

## 🙏 Acknowledgments

- [Stable Diffusion](https://github.com/CompVis/stable-diffusion) for the base model
- [Hugging Face](https://huggingface.co/) for model hosting and Spaces
- [Streamlit](https://streamlit.io/) for the web interface