Update README.md

README.md

@@ -12,24 +12,39 @@ widget:
 - text: a pkblz ball in the middle of a miniature jungle
 ---
 
-# DreamBooth model for the pkblz (pokeball) concept trained by simonschoe on a dataset of basic pokeball (the plain white-red one) images.
+# The Pokeball Machine
 
-This is a Stable Diffusion model fine-tuned on the pkblz (pokeball) concept with DreamBooth. It can be used by modifying the `instance_prompt`: **a pkblz ball in the middle of a miniature jungle**
+The **Pokeball Machine** is a Dreambooth model for the `pokeball` concept (represented by the `pkblz` identifier).
+It applies to the *wildcard* theme.
+It is fine-tuned from `CompVis/stable-diffusion-v1-4` checkpoint on a small dataset of pokeball images (i.e., images of the red-white original pokeball).
+It can be used by modifying the `instance_prompt`: **a pkblz ball in the middle of a miniature jungle**
 
 This model was created as part of the DreamBooth Hackathon 🔥. Visit the [organisation page](https://huggingface.co/dreambooth-hackathon) for instructions on how to take part!
 
-**Training Details:**
+#### Fine-Tuning Details
 - Number of training images: 31
 - Learning rate: 2e-06
 - Training steps: 800
 - Guidance Scale: 10
-- Inference Steps: 75
+- Inference Steps: 50-75
+
+#### Fine-Tuning Examples
 
 <table>
   <tr>
-    <td>Prompt</td>
-    <td>Prompt</td>
-    <td>Prompt</td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/input/pokeball%20(1).jpg" style="height:250px"> </td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/input/pokeball%20(2).jpg" style="height:250px"> </td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/input/pokeball%20(3).jpg" style="height:250px"> </td>
+  </tr>
+</table>
+
+#### Output Examples
+
+<table>
+  <tr>
+    <td>a blueprint photo of a <b>pkblz</b> ball</td>
+    <td>a photo of a cybernetic <b>pkblz</b> ball, wide shot</td>
+    <td>a photo of a <b>pkblz</b> ball in the style vintage disney</td>
   </tr>
   <tr>
     <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(1).png" style="height:250px"> </td>
@@ -37,9 +52,9 @@ This model was created as part of the DreamBooth Hackathon 🔥. Visit the [orga
     <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(3).png" style="height:250px"> </td>
   </tr>
   <tr>
-    <td>Prompt</td>
-    <td>Prompt</td>
-    <td>Prompt</td>
+    <td>a photo of a mosaic <b>pkblz</b> ball lying in an antique temple</td>
+    <td>a photo of a detailed ornate <b>pkblz</b> ball</td>
+    <td>a pkblz ball underwater</td>
   </tr>
   <tr>
     <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(4).png" style="height:250px"> </td>
@@ -47,25 +62,48 @@ This model was created as part of the DreamBooth Hackathon 🔥. Visit the [orga
     <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(6).png" style="height:250px"> </td>
   </tr>
   <tr>
-    <td>Prompt</td>
-    <td>Prompt</td>
-    <td>Prompt</td>
+    <td>a <b>pkblz</b> ball in the middle of a miniature jungle</td>
+    <td>a <b>pkblz</b> ball underwater</td>
+    <td>a mystic <b>pkblz</b> ball, trending on artstation</td>
   </tr>
   <tr>
     <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(7).png" style="height:250px"> </td>
     <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(8).png" style="height:250px"> </td>
     <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(9).png" style="height:250px"> </td>
   </tr>
+    <tr>
+    <td>a <b>pkblz</b> ball underwater, trending on artstation</td>
+    <td>a wooden <b>pkblz</b> ball</td>
+    <td>a <b>pkblz</b> ball hovering over a pond</td>
+  </tr>
+  <tr>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(10).png" style="height:250px"> </td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(11).png" style="height:250px"> </td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(12).png" style="height:250px"> </td>
+  </tr>
+    <tr>
+    <td>a <b>pkblz</b> ball on a sunny tropical beach</td>
+    <td>a steampunk <b>pkblz</b> ball, trending on artstation</td>
+    <td>a colored pencil sketch of a <b>pkblz</b> ball</td>
+  </tr>
+  <tr>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(13).png" style="height:250px"> </td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(14).png" style="height:250px"> </td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(15).png" style="height:250px"> </td>
+  </tr>
+    <tr>
+    <td>a photo of a spectral ornate <b>pkblz</b> ball, trending on artstation, realistic</td>
+    <td>a sunset photo of a <b>pkblz</b> ball</td>
+    <td>a watercolor photo of a <b>pkblz</b> ball</td>
+  </tr>
+  <tr>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(16).png" style="height:250px"> </td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(17).png" style="height:250px"> </td>
+    <td align="center"><img src="https://huggingface.co/simonschoe/pokeball-machine/resolve/main/output/pokeball%20(18).png" style="height:250px"> </td>
+  </tr>
 </table>
 
 
-
-## Description
-
-
-This is a Stable Diffusion model fine-tuned on `(poke)ball` images for the wildcard theme.
-
-
 ## Usage
 
 ```python