Update README.md
Browse files
README.md
CHANGED
@@ -11,3 +11,4 @@ In the "Inference API" interface on this page:
|
|
11 |
- LABEL_0 is a NORMAL result
|
12 |
- LABEL_1 is a positive result for PNEUMONIA
|
13 |
|
|
|
|
11 |
- LABEL_0 is a NORMAL result
|
12 |
- LABEL_1 is a positive result for PNEUMONIA
|
13 |
|
14 |
+
*Model Architecture: This model is based on Google’s ViT (Vision Transformer), fine-tuned for image classification with two labels. The original ViT model was implemented using the google/vit-base-patch16-224 architecture from Hugging Face's Transformers library.
|