Update README.md

README.md

@@ -5,6 +5,15 @@ tags:
 model-index:
 - name: SkLIP-masked
   results: []
+license: mit
+datasets:
+- joshuachou/SkinCAP
+language:
+- en
+base_model:
+- allenai/scibert_scivocab_uncased
+- openai/clip-vit-base-patch32
+pipeline_tag: feature-extraction
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -13,7 +22,7 @@ should probably proofread and complete it, then remove this comment. -->
 # SkLIP
 
 SkLIP (Skin CLIP) is a hybrid CLIP model finetuned on the [SkinCAP](https://hf.rst.im/datasets/joshuachou/SkinCAP), a multi-modal dermatology dataset annotated 
-with rich medical captions. It is built witha RoBERTa text encoder and the pre-trained CLIP vision encoder.
+with rich medical captions. It is built witha SciBERT text encoder and the pre-trained CLIP-32 vision encoder.
 
 ## Model description
 
@@ -62,4 +71,4 @@ The following hyperparameters were used during training:
 - Transformers 4.45.2
 - Pytorch 2.1.0+cu118
 - Datasets 3.0.1
-- Tokenizers 0.20.1
+- Tokenizers 0.20.1