Update README.md

README.md

@@ -1,5 +1,6 @@
 ---
 base_model: black-forest-labs/FLUX.1-dev
+datasets: TIGER-Lab/OmniEdit-Filtered-1.2M
 library_name: diffusers
 license: other
 inference: true
@@ -10,14 +11,29 @@ tags:
 - diffusers
 - control
 - diffusers-training
+widget:
+- text: Give this the look of a traditional Japanese woodblock print.
+  output:
+    url: >-
+      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/edited_car.jpg
+- text: transform the setting to a winter scene
+  output:
+    url: >-
+      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/green_creature.jpg
+- text: turn the color of mushroom to gray
+  output:
+    url: >-
+      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/mushroom.jpg
+- text: Change it to look like it's in the style of an impasto painting.
+  output:
+    url: >-
+      https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/flux-edit-artifacts/norte_dam.jpg
 ---
 
-<!-- This model card has been generated automatically according to the information the training script had access to. You
-should probably proofread and complete it, then remove this comment. -->
-
-
 # Flux Edit
 
+<Gallery />
+
 These are the control weights trained on [black-forest-labs/FLUX.1-dev](htpss://hf.co/black-forest-labs/FLUX.1-dev) 
 and [TIGER-Lab/OmniEdit-Filtered-1.2M](https://huggingface.co/datasets/TIGER-Lab/OmniEdit-Filtered-1.2M) for image editing. We use the
 [Flux Control framework](https://blackforestlabs.ai/flux-1-tools/) for fine-tuning.