Update README.md

README.md

@@ -16,33 +16,74 @@ inference: true
 
 This pipeline was finetuned from **runwayml/stable-diffusion-v1-5** on the **iamkaikai/amazing_logos_v4** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['Simple elegant logo for Grupo Altair Publicidad, Circle Lines Venezuela, Publishing, successful vibe, minimalist, thought-provoking, abstract, recognizable, relatable, sharp, vector art, even edges']: 
 
-![val_imgs_grid](./val_imgs_grid.png)
-
-
-## Pipeline usage
-
-You can use the pipeline like so:
-
-```python
-from diffusers import DiffusionPipeline
-import torch
-
-pipeline = DiffusionPipeline.from_pretrained("iamkaikai/amazing-logos-v4", torch_dtype=torch.float16)
-prompt = "Simple elegant logo for Grupo Altair Publicidad, Circle Lines Venezuela, Publishing, successful vibe, minimalist, thought-provoking, abstract, recognizable, relatable, sharp, vector art, even edges"
-image = pipeline(prompt).images[0]
-image.save("my_image.png")
-```
 
 ## Training info
 
 These are the key hyperparameters used during training:
 
-* Epochs: 3
-* Learning rate: 5e-07
+* Dataset size: 400k
+* Epochs: 2
+* Learning rate: 1e-07
 * Batch size: 1
 * Gradient accumulation steps: 1
 * Image resolution: 512
 * Mixed-precision: fp16
 
 
-More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/iam-kai-kai/text2image-fine-tune/runs/m8nx79k9).
+
+![preview](./grid.jpeg)
+
+
+## Prompt Format
+The prompt format is as follows:
+
+```javascript
+{template keywords} + [company name] + [concept & country] + [industry] + {template keywords}
+```
+
+## For example:
+
+Colored logo
+```text
+Simple elegant logo for Google, G circle United states, technology, successful vibe, minimalist, thought-provoking, abstract, recognizable
+```
+monochrome logo (include "black and white")
+```text
+Simple elegant logo for Google, G circle United states, technology, successful vibe, minimalist, thought-provoking, abstract, recognizable, black and white
+```
+
+Here are some examples of prompts:
+
+- Simple elegant logo for Digital Art, **D A square**, **education**, successful vibe, minimalist, thought-provoking, abstract, recognizable
+- Simple elegant logo for 3M Technology, **3 M square United states**, **technology and product**, successful vibe, minimalist, thought-provoking, abstract, recognizable
+- Simple elegant logo for 42Studio, **lines drop fire flame water**, **design studio**, successful vibe, minimalist, thought provoking, abstract, recognizable, relatable, sharp, vector art, even edges
+
+
+## The [concept & country] section can include words such as:
+- lines
+- circles
+- triangles
+- dot
+- crosses
+- waves
+- square
+- letters (A-Z)
+- 3D
+- Angled
+- Arrows
+- cube
+- Diamond
+- Hexagon
+- Loops
+- outline
+- ovals
+- rectangle
+- reflection
+- rings
+- round
+- semicircle
+- spiral
+- woven
+- stars
+
+I invest in cloud GPU rentals to train my models. If you value what I do, consider buying me a coffee ☕️. Your support means a lot! https://www.buymeacoffee.com/iamkaikai666