Update README.md

README.md

@@ -14,11 +14,21 @@ The model is compatible with Transformers:
 
 ```python
 from transformers import AutoModel, AutoProcessor
+from PIL import Image
+import torch
 
-repo_id = "jlee-larr/dynaflip-base"
+REPO = "jlee-larr/dynaflip-base"
+dynaflip = AutoModel.from_pretrained(REPO, trust_remote_code=True).eval()
+processor = AutoProcessor.from_pretrained(REPO, trust_remote_code=True)
 
-model = AutoModel.from_pretrained("jlee-larr/dynaflip-base")
-processor = AutoProcessor.from_pretrained("jlee-larr/dynaflip-base")
+image = Image.open("example.png").convert("RGB")
+inputs = processor(images=image, return_tensors="pt")
+
+with torch.no_grad():
+    v = dynaflip.vision_outputs(inputs["pixel_values"])
+
+# v.last_hidden_state  -> (B, num_patches, 768)   patch tokens
+# v.pooler_output      -> (B, 1536)               CLS + mean(patches)
 ```
 
 ## Citation