Update README.md

README.md

@@ -42,7 +42,7 @@ device = torch.device("cuda:1" if torch.cuda.is_available() else "cpu")
 device = 'cpu'
 
 # load model
-# If you cannot connect to huggingface.co, you can download the repo and set from_pretrained path as the loacl dir path, replacing "yuxindu/segvol"
+# IF you cannot connect to huggingface.co, you can download the repo and set from_pretrained path as the loacl dir path, replacing "yuxindu/segvol"
 clip_tokenizer = AutoTokenizer.from_pretrained("yuxindu/segvol")
 model = AutoModel.from_pretrained("yuxindu/segvol", trust_remote_code=True, test_mode=True)
 model.model.text_encoder.tokenizer = clip_tokenizer
@@ -59,6 +59,7 @@ categories = ["liver", "kidney", "spleen", "pancreas"]
 
 # generate npy data format
 ct_npy, gt_npy = model.processor.preprocess_ct_gt(ct_path, gt_path, category=categories)
+# IF you have download our 25 processed datasets, you can skip to here with the processed ct_npy, gt_npy files
 
 # go through zoom_transform to generate zoomout & zoomin views
 data_item = model.processor.zoom_transform(ct_npy, gt_npy)