Hyunil commited on
Commit
cf3fc66
·
verified ·
1 Parent(s): 07d6957

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -33,21 +33,21 @@ import torch
33
  from datasets import load_dataset
34
  from transformers import AutoImageProcessor, AutoModelForImageClassification
35
 
36
- # 예시 데이터: 고양이 이미지
37
  dataset = load_dataset("huggingface/cats-image")
38
  image = dataset["test"]["image"][0]
39
 
40
- # 👉 CSATv2 모델로 교체
41
  model_name = "Hyunil/CSATv2"
42
 
43
- # Preprocessor + Model 로드
44
  processor = AutoImageProcessor.from_pretrained(model_name, trust_remote_code=True)
45
  model = AutoModelForImageClassification.from_pretrained(model_name, trust_remote_code=True)
46
 
47
- # 전처리
48
  inputs = processor(image, return_tensors="pt")
49
 
50
- # 추론
51
  with torch.no_grad():
52
  logits = model(**inputs).logits
53
 
 
33
  from datasets import load_dataset
34
  from transformers import AutoImageProcessor, AutoModelForImageClassification
35
 
36
+ # example data : cat image
37
  dataset = load_dataset("huggingface/cats-image")
38
  image = dataset["test"]["image"][0]
39
 
40
+ # 👉 CSATv2
41
  model_name = "Hyunil/CSATv2"
42
 
43
+ # Preprocessor + Model Load
44
  processor = AutoImageProcessor.from_pretrained(model_name, trust_remote_code=True)
45
  model = AutoModelForImageClassification.from_pretrained(model_name, trust_remote_code=True)
46
 
47
+ # Preprocessing
48
  inputs = processor(image, return_tensors="pt")
49
 
50
+ # Inference
51
  with torch.no_grad():
52
  logits = model(**inputs).logits
53