Update README.md

README.md

@@ -96,23 +96,9 @@ model-index:
 pipeline_tag: feature-extraction
 tags:
 - mteb
+- sentence-transformers
 ---
 ```python
-import torch.nn as nn
 from sentence_transformers import SentenceTransformer
-from modeling_adaptor import MixtureOfAdaptors
-class CustomSentenceTransformer(nn.Module):
-    def __init__(self, output_dim=1536):
-        super(CustomSentenceTransformer, self).__init__()
-        self.model = SentenceTransformer('TencentBAC/Conan-embedding-v1', trust_remote_code=True)
-        adaptor = MixtureOfAdaptors(5, 1792)
-        adaptor.load_state_dict(torch.load(f"conan-adaptors.pth"))
-        self.model.add_module('adaptor', adaptor)
-        self.output_dim = output_dim
-    
-    def encode(self, sentences, **kwargs):
-        embeddings = self.model.encode(sentences, **kwargs)
-        return embeddings[:, :self.output_dim]
-
-model = CustomSentenceTransformer(output_dim=1536)
+model = SentenceTransformer("OpenSearch-AI/Ops-MoA-Conan-embedding-v1")
 model.encode(['text'])