Update README.md

README.md

@@ -14,18 +14,16 @@ pipeline_tag: image-classification
 
 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:
-  -- 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
+A reduced version of 6DRepNet model using the backbone of RepVGG A0 backbone
 ---
 
 model-index:
   - name: 6DRepNet-RepVGGA0
     results:
       - task:
-          type: head pose estimation
+          type: Image-Classification
         dataset:
           name: BIWI
           type: Benchmarking dataset