Update src/similarity/similarity.py

src/similarity/similarity.py

@@ -12,6 +12,7 @@ class Similarity:
             model.SimilarityModel(name= 'Mobilenet V3', image_size= 224, model_cls = ModelnetV3()),
             model.SimilarityModel(name= 'Big Transfer (BiT)', image_size= 224, model_cls = BigTransfer()),
             model.SimilarityModel(name= 'Vision Transformer', image_size= 224, model_cls = VitBase(), image_input_type='pil'),
+            model.SimilarityModel(name= 'Masked Siamese Networks for Label-Efficient Learning', image_size= 224, model_cls = ViTMS(), image_input_type='pil')
             ]        
 
     def check_similarity(self, img_urls, model):