Update README.md

README.md

@@ -17,7 +17,8 @@ library_name: pytorch
 This model was created as a part of the project for FI:PA228 (Masaryk University), 
 inspired by this paper: [Face Generation from Textual Features using Conditionally trained Inputs to Generative Adversarial Networks](https://arxiv.org/abs/2301.09123)
 
-It was trained on the generated dataset from BLIP and StyleGAN3.
+It was trained on the generated dataset from BLIP and StyleGAN3. See the [corresponding notebook](https://colab.research.google.com/drive/14rDcCc0Xr1L1Ax3aKezEhmcn81vXGVQ7?usp=sharing)
+for further details.
 
 
 ## How to use:
@@ -52,4 +53,16 @@ model.load_state_dict(torch.load(TRANSLATION_MODEL, map_location=dev))
 ## Demo
 
 For the demo of the whole pipeline, or how this model helps to generate a final image,
-visits [text-to-stylegan HF space](https://huggingface.co/spaces/balgot/text-to-stylegan3).
+visits [text-to-stylegan HF space](https://huggingface.co/spaces/balgot/text-to-stylegan3).
+
+## Examples
+
+* Prompt: `attractive young woman, blond hair`
+  ![image of attractive young women](attractive_young_woman_blonde.png)
+
+
+  
+
+* Prompt initial: `cute young boy, blond hair, blue eyes, smiling`
+* Prompt second: `old man, short gray hair, glasses, wearing hat`
+  <img src="https://huggingface.co/balgot/bert-2-stylegan3/resolve/main/young2old.gif" width="200" height="200" />