Spaces:

SeyedAli
/

Image-Similarity

Runtime error

SeyedAli commited on Sep 29, 2023

Commit

0dc5184

•

1 Parent(s): 6ccb40c

Update src/similarity/similarity.py

Files changed (1) hide show

src/similarity/similarity.py CHANGED Viewed

@@ -3,6 +3,7 @@ from src.util import image as image_util
 from src.util import matrix
 from .model_implements.mobilenet_v3 import ModelnetV3
 from .model_implements.vit_base import VitBase
 from .model_implements.bit import BigTransfer

 from src.util import matrix
 from .model_implements.mobilenet_v3 import ModelnetV3
 from .model_implements.vit_base import VitBase
+from .model_implements.vit_base import ViTMS
 from .model_implements.bit import BigTransfer