cisco-ai
/

SecureBERT2.0-biencoder

Sentence Similarity

sentence-transformers

feature-extraction

Generated from Trainer

dataset_size:35705

loss:MultipleNegativesRankingLoss

text-embeddings-inference

Model card Files Files and versions

cisco-ehsan commited on Oct 6

Commit

b040c8c

·

verified ·

1 Parent(s): 1467022

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -113,8 +113,8 @@ from sentence_transformers import CrossEncoder, util
 from transformers import AutoTokenizer, TFAutoModel
 # Load the bi-encoder
-tokenizer = AutoTokenizer.from_pretrained("/teamspace/studios/this_studio/secure_modern_bert/Models/reranker-securebert2-biencoder/final")
-model = TFAutoModel.from_pretrained("/teamspace/studios/this_studio/secure_modern_bert/Models/reranker-securebert2-biencoder/final")
 # Encode queries and documents
 query_embedding = model(tokenizer("Example query", return_tensors="tf"))[0]
@@ -173,7 +173,6 @@ print(similarity)
 - Datasets: 3.6.0
 - Tokenizers: 0.21.1
-## Citation
 ## Reference

 from transformers import AutoTokenizer, TFAutoModel
 # Load the bi-encoder
+tokenizer = AutoTokenizer.from_pretrained("CiscoAITeam/SecureBERT2.0-biencoder"")
+model = TFAutoModel.from_pretrained("CiscoAITeam/SecureBERT2.0-biencoder"")
 # Encode queries and documents
 query_embedding = model(tokenizer("Example query", return_tensors="tf"))[0]
 - Datasets: 3.6.0
 - Tokenizers: 0.21.1
 ## Reference