change_folder

config.json

@@ -9,8 +9,7 @@
   "head_init_scale": 1.0,
 
   "torch_dtype": "float32",
-
-  // Auto 클래스들이 이 레포 안의 어떤 코드를 써야 하는지 알려주는 부분
+  
   "auto_map": {
     "AutoConfig": "modeling_csatv2.CSATv2Config",
     "AutoModelForImageClassification": "modeling_csatv2.CSATv2ForImageClassification"

example_2.py

@@ -0,0 +1,16 @@
+from transformers import AutoImageProcessor, AutoModelForImageClassification
+from PIL import Image
+import requests
+
+processor = AutoImageProcessor.from_pretrained("Hyunil/CSATv2", trust_remote_code=True)
+model = AutoModelForImageClassification.from_pretrained("Hyunil/CSATv2", trust_remote_code=True)
+
+url = "https://images.unsplash.com/photo-1516116216624-53e697fedbea"
+image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
+
+inputs = processor(image, return_tensors="pt")
+outputs = model(**inputs)
+probs = outputs.logits.softmax(dim=-1)
+
+top_prob, top_idx = probs.topk(5)
+print(top_idx, top_prob)