Update README.md

README.md

@@ -5,6 +5,13 @@ tags:
 - generated_from_trainer
 metrics:
 - accuracy
+datasets:
+- chest X-rays
+widget:
+- src: https://drive.google.com/uc?id=1yqnhD4Wjt4Y_NGLtijTGGaaw9GL497kQ
+  example_title: PNEUMONIA
+- src: https://drive.google.com/uc?id=1xjcIEDb8kuSd4wF44gCEgsc0PfRvs53m
+  example_title: NORMAL
 model-index:
 - name: vit-finetuned-chest-xray-pneumonia
   results: []
@@ -15,7 +22,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-finetuned-chest-xray-pneumonia
 
-This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the chest-xray-pneumonia dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the [chest-xray-pneumonia](https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.1271
 - Accuracy: 0.9551
@@ -60,7 +67,11 @@ The following hyperparameters were used during training:
 | 0.1063        | 9.0   | 2934 | 0.1473          | 0.9535   |
 | 0.0944        | 10.0  | 3260 | 0.1385          | 0.9583   |
 
-
+## Example Images
+#### Pneumonia Chest X-Ray
+![Pneumonia](https://drive.google.com/uc?id=1yqnhD4Wjt4Y_NGLtijTGGaaw9GL497kQ)
+#### Normal Chest X-Ray
+![Normal](https://drive.google.com/uc?id=1xjcIEDb8kuSd4wF44gCEgsc0PfRvs53m)
 ### Framework versions
 
 - Transformers 4.17.0