Update README.md

README.md

@@ -1,11 +1,22 @@
+---
+license: openrail++
+datasets:
+- friedrichor/PhotoChat_120_square_HQ
+language:
+- en
+tags:
+- stable-diffusion
+- text-to-image
+---
 
 fine-tuned with text-image dataset `friedrichor/PhotoChat_120_square_HQ`
 
 # Model Details
 
-- Model type: Diffusion-based text-to-image generation model
-- Language(s): English
-- fine-tuning dataset: [friedrichor/PhotoChat_120_square_HQ](https://huggingface.co/datasets/friedrichor/PhotoChat_120_square_HQ)
+- **Model type:** Diffusion-based text-to-image generation model
+- **Language(s):** English
+- **License:** [CreativeML Open RAIL++-M License](https://huggingface.co/stabilityai/stable-diffusion-2/blob/main/LICENSE-MODEL)
+- **Model Description:** This is a model that can be used to generate and modify images based on text prompts. It is a Latent Diffusion Model that uses a fixed, pretrained text encoder (OpenCLIP-ViT/H). Fine-tuning dataset: [friedrichor/PhotoChat_120_square_HQ](https://huggingface.co/datasets/friedrichor/PhotoChat_120_square_HQ)
 
 ## Dataset
 [friedrichor/PhotoChat_120_square_HQ](https://huggingface.co/datasets/friedrichor/PhotoChat_120_square_HQ) was used for fine-tuning Stable Diffusion v2.1.  
@@ -23,6 +34,8 @@ or [Hugging Face diffusers](https://github.com/huggingface/diffusers/tree/main/e
 
 # Simple use example
 
+Using the [🤗's Diffusers library](https://github.com/huggingface/diffusers)
+
 ```python
 import torch
 from diffusers import StableDiffusionPipeline
@@ -68,5 +81,4 @@ For more prompt templates, see [Dalabad/stable-diffusion-prompt-templates](https
 For example,
 ```
 cartoon, anime, ugly, (aged, white beard, black skin, wrinkle:1.1), (bad proportions, unnatural feature, incongruous feature:1.4), (blurry, un-sharp, fuzzy, un-detailed skin:1.2), (facial contortion, poorly drawn face, deformed iris, deformed pupils:1.3), (mutated hands and fingers:1.5), disconnected hands, disconnected limbs
-```
-
+```