File size: 2,408 Bytes
eff7e87
 
 
 
 
 
 
2c843c7
eff7e87
 
edcf552
eff7e87
 
2c843c7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
---
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