pyh5214 commited on
Commit
eceb8d3
·
verified ·
1 Parent(s): c4f4fe0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -61,8 +61,8 @@ from transformers import AutoModelForImageClassification, AutoFeatureExtractor
61
  import torch
62
 
63
  # 모델과 특징 추출기 불러오기
64
- model = AutoModelForImageClassification.from_pretrained("your-org/ConvNext_Multi")
65
- extractor = AutoFeatureExtractor.from_pretrained("your-org/ConvNext_Multi")
66
 
67
  # 다중밴드 이미지 텐서 (예: [batch_size, 5, H, W])
68
  inputs = extractor(multi_band_images, return_tensors="pt")
 
61
  import torch
62
 
63
  # 모델과 특징 추출기 불러오기
64
+ model = AutoModelForImageClassification.from_pretrained("MhRnd/ConvNext_Multi")
65
+ extractor = AutoFeatureExtractor.from_pretrained("MhRnd/ConvNext_Multi")
66
 
67
  # 다중밴드 이미지 텐서 (예: [batch_size, 5, H, W])
68
  inputs = extractor(multi_band_images, return_tensors="pt")