Update README.md

README.md

@@ -12,3 +12,38 @@ Siglip2-base-patch16-224(https://huggingface.co/google/siglip2-base-patch16-224)
 사용된 평가 데이터 : https://huggingface.co/datasets/hyunlord/aihub_image-caption_en-ko_karpathy-split (test)
 
 ## How to use
+
+```python
+import requests
+import torch
+from PIL import Image
+from transformers import AutoModel, AutoProcessor
+
+repo = "hyunlord/siglip2-base-patch16-224-ko"
+model = AutoModel.from_pretrained(repo)
+processor = AutoProcessor.from_pretrained(repo)
+
+url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
+
+texts = ["고양이 한 마리", 
+         "고양이 두 마리",
+         "분홍색 소파에 드러누운 고양이 친구들",
+         "리모컨과 고양이 두마리",
+         "리모컨 두 개와 고양이 두마리",
+         "분홍색 소파 위에 리모컨 두 개와 드러누운 고양이 두마리"]
+inputs = processor(text=texts,
+                   images=image,
+                   padding="max_length",
+                   max_length=64,
+                   return_tensors="pt")
+with torch.no_grad():
+    outputs = model(**inputs)
+logits_per_image = outputs.logits_per_image
+probs = torch.sigmoid(logits_per_image)
+```
+
+```python
+>>> probs
+tensor([[0.0038, 0.0429, 0.8294, 0.9787, 0.9816, 0.9990]])
+```