Borjamg commited on
Commit
b6c9474
·
verified ·
1 Parent(s): 8bf43bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
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.