Update README.md

README.md

@@ -6,7 +6,31 @@ tags: []
 # Model Card for Model ID
 
 <!-- Provide a quick summary of what the model is/does. -->
-
+Code used to generate the model:
+```py
+import torch
+from transformers import HieraForImageClassification, HieraConfig, AutoProcessor
+
+# Set seed for reproducibility
+torch.manual_seed(0)
+
+# Initializing a Hiera-tiny style configuration
+configuration = HieraConfig(
+    embed_dim=8,
+    hidden_size=32,
+)
+
+# Initializing a model from the Hiera-tiny style configuration
+model = HieraForImageClassification(configuration)
+
+# Re-use hiera-tiny processor
+processor = AutoProcessor.from_pretrained("facebook/hiera-tiny-224-hf")
+
+# Upload to the HF Hub
+model_id = 'hf-internal-testing/tiny-random-HieraForImageClassification'
+model.push_to_hub(model_id)
+processor.push_to_hub(model_id)
+```
 
 
 ## Model Details