Improve model card: Add library_name and correct pipeline_tag (#1)
Browse files- Improve model card: Add library_name and correct pipeline_tag (5d0708955ad28300914e1712bb57aa7ced4e8d76)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
@@ -1,14 +1,16 @@
|
|
1 |
---
|
2 |
-
license: cc-by-nc-4.0
|
3 |
language:
|
4 |
- en
|
5 |
-
|
|
|
|
|
6 |
tags:
|
7 |
- medical
|
8 |
- multimodal
|
9 |
- vision-language pre-training
|
10 |
- chest x-ray
|
11 |
---
|
|
|
12 |
# MeDSLIP: Medical Dual-Stream Language-Image Pre-training for Fine-grained Alignment
|
13 |
|
14 |
## Introduction:
|
@@ -44,7 +46,7 @@ All files for data preparation files can be downloaded from [Hugging Face: MeDSL
|
|
44 |
### Pre-training
|
45 |
Our pre-train code is given in ```PreTrain_MeDSLIP```.
|
46 |
* Check the ```PreTrain_MeDSLIP/data_file``` dir and download the files for data preparation.
|
47 |
-
* Change the data and preparation files paths as you
|
48 |
|
49 |
## Reference
|
50 |
```
|
|
|
1 |
---
|
|
|
2 |
language:
|
3 |
- en
|
4 |
+
license: cc-by-nc-4.0
|
5 |
+
pipeline_tag: image-text-to-text
|
6 |
+
library_name: transformers
|
7 |
tags:
|
8 |
- medical
|
9 |
- multimodal
|
10 |
- vision-language pre-training
|
11 |
- chest x-ray
|
12 |
---
|
13 |
+
|
14 |
# MeDSLIP: Medical Dual-Stream Language-Image Pre-training for Fine-grained Alignment
|
15 |
|
16 |
## Introduction:
|
|
|
46 |
### Pre-training
|
47 |
Our pre-train code is given in ```PreTrain_MeDSLIP```.
|
48 |
* Check the ```PreTrain_MeDSLIP/data_file``` dir and download the files for data preparation.
|
49 |
+
* Change the data and preparation files paths as you desire in ```PreTrain_MeDSLIP/configs/Pretrain_MeDSLIP.yaml```, and ```python PreTrain_MeDSLIP/train_MeDSLIP.py``` to pre-train.
|
50 |
|
51 |
## Reference
|
52 |
```
|