Update README.md

README.md

@@ -24,5 +24,5 @@ model-index:
         name: Val F1 Score    # Optional. Example: Test WER
 ---
 ## Detecting the Orientation of CelebA pictures using Deep Learning
-This model has been trained on a modified version of the CelebA dataset, which was made from flipping 20,000 and keeping 20,000 images intact. 
-The model relies on Resnet-18 as a backbone and then is connected to one output node to classify whether images are flipped upside down (1) or not (0).
+This model has been trained on a modified version of the CelebA-faces dataset, which was made from flipping 20,000 images upside down and keeping 20,000 images intact.<br> 
+The model relies on Resnet-18 as a backbone and is connected to one output node to classify whether the images are flipped upside down (1) or not (0).