Update README.md

README.md

@@ -1,45 +1,155 @@
 ---
 license: creativeml-openrail-m
 tags:
-- text-to-image
-- stable-diffusion
-- art
-- stable-diffusion-diffusers
-- text-to-image
-- diffusers
+  - text-to-image
+  - stable-diffusion
+  - art
+  - stable-diffusion-diffusers
 language:
-- en
+  - en
 pipeline_tag: text-to-image
 library_name: diffusers
 ---
 
-# D.r.e.a.m (Digital Rendering Engine for Artistic Melodies)
-## Welcome to D.r.e.a.m (Digital Rendering Engine for Artistic Melodies).
+<div align="center">
 
-The D.R.E.A.M. model suite is designed for various image generation purposes, featuring specialized models for different styles. The general-purpose image generation model, simply referred to as the D.R.E.A.M. model, caters to a wide range of image creation needs. It is trained on the Cyberverse dataset, ensuring it has a robust understanding of diverse visual contexts and can produce versatile outputs.
+# 🌌 D.R.E.A.M.
+### Digital Rendering Engine for Artistic Melodies
 
-For generating anime-related images, the suite includes the Dream-Anime model. This model is fine-tuned specifically to create high-quality, detailed anime artwork. By leveraging the Surreal Symphonies dataset available on Kaggle, Dream-Anime excels in capturing the distinct stylistic elements and aesthetics of anime, making it an ideal tool for anime enthusiasts and creators.
+*Diffusion-based text-to-image generation, tuned for artistic, anime, and photorealistic rendering.*
 
-Additionally, the Dream-Photorealism model is tailored to generate photorealistic images. Also trained on the Surreal Symphonies dataset, this model focuses on producing images with lifelike quality and precision. Its training enables it to render images that closely mimic real-world visuals, making it suitable for applications requiring high degrees of realism.
+[![Model Type](https://img.shields.io/badge/Model%20Type-Diffusion-8A2BE2?style=for-the-badge)](#)
+[![Library](https://img.shields.io/badge/Library-🤗%20Diffusers-FFD21E?style=for-the-badge)](#)
+[![License](https://img.shields.io/badge/License-CreativeML%20OpenRAIL--M-blue?style=for-the-badge)](#)
+[![Status](https://img.shields.io/badge/Training-In%20Progress-orange?style=for-the-badge)](#)
+[![Language](https://img.shields.io/badge/Language-English-lightgrey?style=for-the-badge)](#)
 
-Each model within the D.R.E.A.M. suite is optimized for its specific purpose, ensuring high performance and quality in its respective domain of image generation.
+</div>
 
-## Model Details
-- **Developed by:** Cyanex1702
-- **Model Type:** Diffusion-based text-to-image generation model
-- **Language(s):** English
-- **Training Status:** In Progress
+---
+
+## 📖 Overview
+
+**D.R.E.A.M.** is a suite of diffusion-based text-to-image models built to emulate the creativity and visual richness of tools like Midjourney — while offering specialized variants for different artistic domains.
+
+> Turn a sentence into a scene. Turn a scene into a style.
+
+---
+
+## 🧩 Model Suite
+
+| Model | Purpose | Training Data | Best For |
+|---|---|---|---|
+| 🎨 **D.R.E.A.M.** (base) | General-purpose image generation | Cyberverse dataset | Versatile, all-around creative outputs |
+| 🌸 **Dream-Anime** | Anime-style artwork | Surreal Symphonies (Kaggle) | Detailed, stylized anime illustrations |
+| 📸 **Dream-Photorealism** | Photorealistic rendering | Surreal Symphonies (Kaggle) | Lifelike, high-fidelity real-world visuals |
+
+---
+
+## 🗺️ How It Works
 
-## Model Description
-D.r.e.a.m is a model designed to generate and modify images based on text prompts. The model leverages advanced diffusion techniques to create high-quality, artistic renderings from textual descriptions, aiming to emulate the style and creativity of Midjourney.
+```
+ 📝 Text Prompt
+       │
+       ▼
+ 🔤 Text Encoder
+       │
+       ▼
+ 🌫️  Diffusion U-Net
+       │
+       ▼
+   Model Variant?
+   ┌───────┼────────────┐
+   ▼       ▼            ▼
+🎨 Base  🌸 Anime   📸 Photorealism
+   │       │            │
+   └───────┴────────────┘
+              │
+              ▼
+       🖼️ Generated Image
+```
+
+---
+
+## ✨ Features
+
+- 🖋️ **Text-to-Image Generation** — Turn descriptive prompts into original images
+- 🛠️ **Image Modification** — Guide edits to existing images with new prompts
+- 🎭 **Creative Rendering** — Artistic, imaginative, Midjourney-inspired outputs
+- 🧬 **Specialized Variants** — Pick the model tuned for your visual style
+
+---
+
+## 📊 Model Comparison
+
+*Illustrative of intended design focus, not benchmarked scores — training is still in progress.*
+
+| Use Case | D.R.E.A.M. (base) | Dream-Anime | Dream-Photorealism |
+|---|---|---|---|
+| General Use | ████████░░ 8/10 | ████░░░░░░ 4/10 | ████░░░░░░ 4/10 |
+| Anime Style | █████░░░░░ 5/10 | █████████░ 9/10 | ███░░░░░���░ 3/10 |
+| Photorealism | █████░░░░░ 5/10 | ███░░░░░░░ 3/10 | █████████░ 9/10 |
+| Prompt Flexibility | █████████░ 9/10 | ██████░░░░ 6/10 | ██████░░░░ 6/10 |
+
+---
+
+## 🚀 Usage
+
+```python
+from diffusers import DiffusionPipeline
+import torch
+
+pipe = DiffusionPipeline.from_pretrained(
+    "Cyanex1702/DREAM",
+    torch_dtype=torch.float16
+).to("cuda")
+
+prompt = "a glowing city floating above the clouds, cinematic lighting, ultra detailed"
+image = pipe(prompt).images[0]
+image.save("dream_output.png")
+```
+
+**Swap in a specialized variant:**
+
+```python
+# Anime style
+pipe = DiffusionPipeline.from_pretrained("Cyanex1702/Dream-Anime", torch_dtype=torch.float16).to("cuda")
+
+# Photorealism
+pipe = DiffusionPipeline.from_pretrained("Cyanex1702/Dream-Photorealism", torch_dtype=torch.float16).to("cuda")
+```
+
+---
+
+## ⚙️ Model Details
+
+| Field | Value |
+|---|---|
+| **Developed by** | [Cyanex1702](https://huggingface.co/Cyanex1702) |
+| **Model type** | Diffusion-based text-to-image generation |
+| **Language(s)** | English |
+| **License** | CreativeML OpenRAIL-M |
+| **Training status** | 🟠 In progress — outputs may vary and contain imperfections |
+
+---
+
+## ⚠️ Limitations
+
+Since training is ongoing, expect:
+- Inconsistent output quality between prompts
+- Occasional artifacts or anatomical errors
+- Style drift on out-of-distribution prompts
+
+---
+
+## 🤝 Contributing
+
+Contributions are welcome! Feel free to open an issue or pull request if you'd like to help improve the D.R.E.A.M. suite — whether that's data, training tips, or example prompts.
+
+---
 
-## Features
-- **Text-to-Image Generation:** Generate images from descriptive text prompts.
-- **Image Modification:** Modify existing images based on new text inputs.
-- **Creative Rendering:** Produce artistic and imaginative images.
+<div align="center">
 
-## Usage
-To use the model, you can input text prompts in English. The model will process these prompts and generate corresponding images. Note that due to the model's current training phase, the results may vary and contain imperfections.
+*Made with 🌙 by Cyanex1702*
 
-## Contributing
-We welcome contributions from the community! If you'd like to contribute.
+</div>