Hugging Face's logo

aehrc
/
cxrmate-rrg24

Feature Extraction
Transformers
Safetensors
vision-encoder-decoder
custom_code
Model card Files
xet
 Community
4
anicolson commited on
Commit
8ca1c27
·
verified ·
1 Parent(s): 42496bf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -33,6 +33,7 @@ import torch
33
  from torchvision.transforms import v2
34
  import transformers
35
  import datasets
 
36
 
37
 
38
  tokenizer = transformers.AutoTokenizer.from_pretrained('aehrc/cxrmate-rrg24')
 
33
  from torchvision.transforms import v2
34
  import transformers
35
  import datasets
36
+ from torch.utils.data import DataLoader
37
 
38
 
39
  tokenizer = transformers.AutoTokenizer.from_pretrained('aehrc/cxrmate-rrg24')