| --- |
| 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 |
|
|
