Update README.md
Browse files
README.md
CHANGED
@@ -20,7 +20,7 @@ from PIL import Image
|
|
20 |
from transformers import BertTokenizer, ViTImageProcessor, VisionEncoderDecoderModel, GenerationConfig
|
21 |
import requests
|
22 |
|
23 |
-
mode = "
|
24 |
# Model
|
25 |
model = VisionEncoderDecoderModel.from_pretrained(f"IAMJB/chexpert-mimic-cxr-{mode}-baseline").eval()
|
26 |
tokenizer = BertTokenizer.from_pretrained(f"IAMJB/chexpert-mimic-cxr-{mode}-baseline")
|
|
|
20 |
from transformers import BertTokenizer, ViTImageProcessor, VisionEncoderDecoderModel, GenerationConfig
|
21 |
import requests
|
22 |
|
23 |
+
mode = "impression"
|
24 |
# Model
|
25 |
model = VisionEncoderDecoderModel.from_pretrained(f"IAMJB/chexpert-mimic-cxr-{mode}-baseline").eval()
|
26 |
tokenizer = BertTokenizer.from_pretrained(f"IAMJB/chexpert-mimic-cxr-{mode}-baseline")
|