Update README.md
Browse files
README.md
CHANGED
@@ -44,10 +44,10 @@ from torchvision.utils import make_grid
|
|
44 |
|
45 |
# Device and paths:
|
46 |
device = 'cuda'
|
47 |
-
physionet_dir = '
|
48 |
-
dataset_dir = '
|
49 |
-
database_path = '
|
50 |
-
mimic_cxr_jpg_dir = '
|
51 |
|
52 |
# Download model checkpoint:
|
53 |
model = transformers.AutoModel.from_pretrained('aehrc/cxrmate-ed', trust_remote_code=True).to(device=device)
|
|
|
44 |
|
45 |
# Device and paths:
|
46 |
device = 'cuda'
|
47 |
+
physionet_dir = '/.../physionet.org/files'
|
48 |
+
dataset_dir = '/.../datasets'
|
49 |
+
database_path = '/.../database/cxrmate_ed.db'
|
50 |
+
mimic_cxr_jpg_dir = '/.../physionet.org/files/mimic-cxr-jpg/2.0.0/files'
|
51 |
|
52 |
# Download model checkpoint:
|
53 |
model = transformers.AutoModel.from_pretrained('aehrc/cxrmate-ed', trust_remote_code=True).to(device=device)
|