Spaces:
Sleeping
Sleeping
File size: 2,019 Bytes
a4aa5c5 463e35b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | """Shared constants for SynthCXR: disease labels, condition maps, severity modifiers."""
from __future__ import annotations
# CheXpert label column names -> natural-language descriptions used in prompts.
# Used by both dataset preparation and inference scripts.
LABEL_TEXT: dict[str, str] = {
"Enlarged Cardiomediastinum": "enlarged cardiomediastinum",
"Cardiomegaly": "cardiomegaly",
"Lung Opacity": "diffuse lung opacity",
"Lung Lesion": "discrete lung lesion",
"Edema": "pulmonary edema",
"Consolidation": "parenchymal consolidation",
"Pneumonia": "findings compatible with pneumonia",
"Atelectasis": "atelectasis",
"Pneumothorax": "pneumothorax",
"Pleural Effusion": "pleural effusion",
"Pleural Other": "other pleural abnormality",
"Fracture": "possible fracture",
"Support Devices": "support devices in place",
}
# Snake_case keys for config files -> natural-language descriptions.
KNOWN_CONDITIONS: dict[str, str] = {
"enlarged_cardiomediastinum": "enlarged cardiomediastinum",
"cardiomegaly": "cardiomegaly",
"lung_opacity": "diffuse lung opacity",
"lung_lesion": "discrete lung lesion",
"edema": "pulmonary edema",
"consolidation": "parenchymal consolidation",
"pneumonia": "findings compatible with pneumonia",
"atelectasis": "atelectasis",
"pneumothorax": "pneumothorax",
"pleural_effusion": "pleural effusion",
"pleural_other": "other pleural abnormality",
"fracture": "possible fracture",
"support_devices": "support devices in place",
}
SEVERITY_MODIFIERS: dict[str, str] = {
"mild": "mild",
"moderate": "moderate",
"severe": "severe",
"small": "small",
"large": "large",
"very_small": "very small",
"very_large": "very large",
"minimal": "minimal",
"significant": "significant",
}
DEFAULT_MODEL_ID = "gradientguild/SynthCXR-Qwen-Weights"
TEXT_ENCODER_MODEL_ID = "gradientguild/SynthCXR-Qwen-Weights"
PROCESSOR_MODEL_ID = "gradientguild/SynthCXR-Qwen-Weights"
|