Update README.md

README.md

@@ -10,6 +10,9 @@ metrics:
 model-index:
 - name: aesthetic_attribute_classifier
   results: []
+widget:
+- text: Check your vertical on the main support; it looks a little off. I'd also like to see how it looks with a bit of the sky cropped from the photo
+
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,7 +20,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # aesthetic_attribute_classifier
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the [PCCD dataset](https://github.com/ivclab/DeepPhotoCritic-ICCV17).
 It achieves the following results on the evaluation set:
 - Loss: 0.3976
 - Precision: {'precision': 0.877129341279301}