Update README.md

README.md

@@ -21,3 +21,22 @@ metrics:
 
 - pre-trained: 0.4759327471256256
 - fine-tuned: 0.9957262277603149
+
+## Usage
+
+load vision model
+
+```python
+from transformers import CLIPVisionModel
+
+vision_model = CLIPVisionModel.from_pretrained('tanganke/clip-vit-base-patch32_mnist')
+```
+
+substitute the vision encoder of clip
+
+```python
+from transformers import CLIPModel
+
+clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
+clip_model.vision_model.load_state_dict(vision_model.vision_model.state_dict())
+```