Update README.md

README.md

@@ -14,7 +14,8 @@ base_model:
 pipeline_tag: image-segmentation
 ---
 
-This model performs both segmentation and classification on chest radiographs (X-rays). 
+This model performs both segmentation and classification on chest radiographs (X-rays).
+The model uses a `tf_efficientnetv2_s` backbone with a U-Net decoder for segmentation and linear layer for classification.
 For frontal radiographs, the model segments the: 1) right lung, 2) left lung, and 3) heart. 
 The model also predicts the chest X-ray view (AP, PA, lateral), patient age, and patient sex.
 The [CheXpert](https://stanfordmlgroup.github.io/competitions/chexpert/) (small version) and [NIH Chest X-ray](https://nihcc.app.box.com/v/ChestXray-NIHCC) datasets were used to train the model.