anicolson commited on
Commit
d002093
·
verified ·
1 Parent(s): a8d7a4a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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])]