Update README.md

README.md

@@ -6,7 +6,35 @@ tags: []
 # Model Card for Model ID
 
 <!-- Provide a quick summary of what the model is/does. -->
-
+## Code to create
+```python
+import torch
+from transformers import VitPoseConfig, VitPoseForPoseEstimation, VitPoseBackboneConfig, AutoProcessor
+
+model_id = "usyd-community/vitpose-plus-small"
+
+# Initializing a VitPose configuration
+configuration = VitPoseConfig.from_pretrained(
+    model_id,
+    backbone_config=VitPoseBackboneConfig(
+        hidden_size=16,
+        num_experts=2,
+        num_attention_heads=2,
+        part_features=10,
+    ),
+)
+
+# Initializing a model (with random weights) from the configuration
+model = VitPoseForPoseEstimation(configuration)
+
+torch.manual_seed(0) # Set for reproducibility
+for name, param in model.named_parameters():
+    param.data = torch.randn_like(param)
+
+# Accessing the model configuration
+configuration = model.config
+processor = AutoProcessor.from_pretrained(model_id)
+```
 
 
 ## Model Details