Update README.md

README.md

@@ -2,8 +2,30 @@
 tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
+license: apache-2.0
+language:
+- en
+metrics:
+- mae
 ---
 
 This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
-- Library: [More Information Needed]
-- Docs: [More Information Needed]
+- Library:
+  torch==2.2.1+cu118
+  torchaudio==2.2.1+cu118
+  torchvision==0.17.1
+- Docs:
+  A reduced version of 6DRepNet model using the backbone of RepVGG A0 backbone
+model-index:
+  - name: 6DRepNet-RepVGGA0
+    results:
+      - task:
+          type: head pose estimation
+        dataset:
+          name: BIWI
+          type: Benchmarking
+        metrics:
+          - name: MAE
+            type: MAE
+            value: 3.70
+            verified: false