Update README.md

README.md

@@ -8,12 +8,15 @@ tags:
 - prompt generator
 - stable-diffusion
 - mistralai/Mistral-7B-v0.1
-- StableDiffusionXLPipeline
 - 'inference: false'
 ---
 
-# Prompt Diffuser
+<br>
+
+<h1 style="font-size: 2em;">Prompt Diffuser</h1>
 
+| <img src="coffee.png" width="400"/> | <img src="skii.png" width="400"/> | <img src="dance_rain.png" width="400"/> | <img src="man_beach.png" width="400"/> |
+|-------------------------------------|----------------------------------|-----------------------------------------|--------------------------------------|
 
 
 This model was build via parameter-efficient QLoRA finetuning of [Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on a small custom dataset.
@@ -217,9 +220,15 @@ prompt = generate_prompt(input_text, 150, 0.3)
 print("\nPrompt: " + prompt)
 ```
 
+## Contributing
+
+Contributions are welcome! If you find any bugs, create an issue or submit a pull request with your proposed changes.
+
 ## Acknowledgements
 
-This model was finetuned by Abhishek Kalra on Sep 29, 2023 and is for research applications only.
+This model was finetuned by [Abhishek Kalra](https://github.com/abhishek7kalra) on Sep 29, 2023 and is for research applications only.
+
+[![Gmail](https://img.shields.io/badge/Gmail-D14836?style=for-the-badge&logo=gmail&logoColor=white)](mailto:abhishek7.kalra@gmail.com)
 
 ## mistralai/Mistral-7B-v0.1 citation