MohamedRashad
commited on
Commit
•
b1c4379
1
Parent(s):
655c0ac
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,24 @@
|
|
1 |
---
|
2 |
license: openrail
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: openrail
|
3 |
---
|
4 |
+
|
5 |
+
# Stable Diffusion fine-tuned for [Fashion Product Images Dataset](https://www.kaggle.com/datasets/paramaggarwal/fashion-product-images-dataset)
|
6 |
+
|
7 |
+
This model is a fine-tuned version of [openjourney](https://huggingface.co/prompthero/openjourney) that is based on Stable Diffusion targeting fashion and clothing.
|
8 |
+
|
9 |
+
## How to use ?
|
10 |
+
|
11 |
+
```python
|
12 |
+
from diffusers import StableDiffusionPipeline
|
13 |
+
import torch
|
14 |
+
|
15 |
+
pipeline = StableDiffusionPipeline.from_pretrained("MohamedRashad/diffusion_fashion", torch_dtype=torch.float16)
|
16 |
+
pipeline.to("cuda")
|
17 |
+
|
18 |
+
prompt = "A photo of a dress, made in 2019, color is Red, Casual usage, Women's cloth, something for the summer season, on white background"
|
19 |
+
images = pipeline(prompt).images[0]
|
20 |
+
image.save("red_dress.png")
|
21 |
+
|
22 |
+
```
|
23 |
+
|
24 |
+
## Any feedback or questions are welcomed on the [community](https://huggingface.co/MohamedRashad/diffusion_fashion/discussions) tab
|