Update README.md

README.md

@@ -1,3 +1,28 @@
 ## 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).
+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).
+---
+language:
+- en 
+license: apache-2.0  
+tags:
+- {image-classification}  
+- {pytorch}  
+datasets:
+- CelebA-faces 
+
+model-index:
+- name: celebA_orientation_detection_model
+  results:
+  - task: 
+      type: image_classification # Required. Example: automatic-speech-recognition
+      name: Image Classification # Optional. Example: Speech Recognition
+    dataset:
+      type: CelebA-faces   
+      name: CelebA-faces   
+      
+    metrics:
+      - type: f1score    # Required. Example: wer
+        value: 0.97  # Required. Example: 20.90
+        name: Val F1 Score    # Optional. Example: Test WER
+---