Improve model card: Add library_name and correct pipeline_tag
Browse filesThis PR improves the model card by:
- Correcting the `pipeline_tag` to `image-text-to-text` to accurately reflect the model's functionality.
- Adding the `library_name` to the metadata for improved discoverability.
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 |
```
|