Update README.md

README.md

@@ -30,16 +30,18 @@ cd unicom/mlcd
 
 ```python
 from vit_rope2d_hf import MLCDVisionModel
-from transformers import AutoImageProcessor
+from transformers import CLIPImageProcessor
 from PIL import Image
+import requests
 import torch
 
 # Load model and processor
 model = MLCDVisionModel.from_pretrained("DeepGlint-AI/mlcd-vit-bigG-patch14-336")
-processor = AutoImageProcessor.from_pretrained("DeepGlint-AI/mlcd-vit-bigG-patch14-336")
+processor = CLIPImageProcessor.from_pretrained("DeepGlint-AI/mlcd-vit-bigG-patch14-336")
 
 # Process single image
-image = Image.open("document.jpg").convert("RGB")
+url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
 inputs = processor(images=image, return_tensors="pt")
 
 # Get visual features
@@ -48,4 +50,5 @@ with torch.no_grad():
 features = outputs.last_hidden_state
 
 print(f"Extracted features shape: {features.shape}")
+# Extracted features shape: torch.Size([1, 577, 1664])
 ```