Update README.md
Browse files
README.md
CHANGED
@@ -95,10 +95,10 @@ from torchvision.io import read_image
|
|
95 |
model = transformers.AutoModelForCausalLM.from_pretrained('aehrc/cxrmate-ed', trust_remote_code=True).to(device=device)
|
96 |
tokenizer = transformers.PreTrainedTokenizerFast.from_pretrained('aehrc/cxrmate-ed')
|
97 |
|
98 |
-
study_image_paths = ['...', '...']
|
99 |
|
100 |
-
indication = '...'
|
101 |
-
history = '...'
|
102 |
|
103 |
images = [read_image(i) for i in img_path_list_idx]
|
104 |
images = [torch.stack([model.test_transforms(i) for i in images])]
|
|
|
95 |
model = transformers.AutoModelForCausalLM.from_pretrained('aehrc/cxrmate-ed', trust_remote_code=True).to(device=device)
|
96 |
tokenizer = transformers.PreTrainedTokenizerFast.from_pretrained('aehrc/cxrmate-ed')
|
97 |
|
98 |
+
study_image_paths = ['...', '...'] # e.g., ['img1.jpeg', 'img2.jpeg'].
|
99 |
|
100 |
+
indication = '...' # Set to None if not using.
|
101 |
+
history = '...' # Set to None if not using.
|
102 |
|
103 |
images = [read_image(i) for i in img_path_list_idx]
|
104 |
images = [torch.stack([model.test_transforms(i) for i in images])]
|