Update README.md

README.md

@@ -5,6 +5,7 @@ tags:
 model-index:
 - name: distilgpt2-sd-prompts
   results: []
+
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -12,7 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilgpt2-sd-prompts
 
-This model is a fine-tuned version of [distilgpt2](https://huggingface.co/distilgpt2) on an unknown dataset.
+This model is a fine-tuned version of [distilgpt2](https://huggingface.co/distilgpt2) on [Stable-Diffusion-Prompts](https://huggingface.co/datasets/Gustavosta/Stable-Diffusion-Prompts).
 It achieves the following results on the evaluation set:
 - Loss: 0.9450