Upload README.md with huggingface_hub

README.md

@@ -1,6 +1,7 @@
 
 ---
 license: creativeml-openrail-m
+base_model: runwayml/stable-diffusion-v1-5
 datasets:
 - Drozdik/tattoo_v3
 tags:
@@ -11,12 +12,11 @@ tags:
 inference: true
 ---
     
-# Text-to-image diffusion - TejasNavada/tattoo-diffusion
+# Text-to-image finetuning - TejasNavada/tattoo-diffusion
 
-This pipeline was trained on the **Drozdik/tattoo_v3** dataset. Below are some example images generated with the finetuned pipeline using the following prompts:           
-['a dragon on a white background', ' a fiery skull', 'a skull', 'a face', 'a snake and skull']
+This pipeline was finetuned from **runwayml/stable-diffusion-v1-5** on the **Drozdik/tattoo_v3** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['a dragon on a white background', ' a fiery skull', 'a skull', 'a face', 'a snake and skull']: 
 
-![val_imgs_grid](./samples/0029.png)
+![val_imgs_grid](./val_imgs_grid.png)
 
 
 ## Pipeline usage
@@ -37,7 +37,7 @@ image.save("my_image.png")
 
 These are the key hyperparameters used during training:
 
-* Epochs: 30
+* Epochs: 50
 * Learning rate: 1e-05
 * Batch size: 2
 * Image resolution: 512