Sentence Similarity
sentence-transformers
Safetensors
Transformers
English
bert
feature-extraction
text-embeddings-inference
Instructions to use Pavithiran/embeddings with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use Pavithiran/embeddings with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("Pavithiran/embeddings") sentences = [ "That is a happy person", "That is a happy dog", "That is a very happy person", "Today is a sunny day" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use Pavithiran/embeddings with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("Pavithiran/embeddings") model = AutoModel.from_pretrained("Pavithiran/embeddings") - Notebooks
- Google Colab
- Kaggle
Update model.py
Browse files
model.py
CHANGED
|
@@ -7,13 +7,9 @@ class Model:
|
|
| 7 |
self.embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
|
| 8 |
|
| 9 |
def __call__(self, payload):
|
| 10 |
-
# Extract
|
| 11 |
-
|
| 12 |
-
source_sentence = inputs.get("source_sentence", "")
|
| 13 |
-
sentences = inputs.get("sentences", [])
|
| 14 |
|
| 15 |
-
# Combine source_sentence with sentences
|
| 16 |
-
chunks = [source_sentence] + sentences
|
| 17 |
# Generate embeddings
|
| 18 |
embeddings = self.embedding_model.encode(chunks, convert_to_tensor=True)
|
| 19 |
|
|
|
|
| 7 |
self.embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
|
| 8 |
|
| 9 |
def __call__(self, payload):
|
| 10 |
+
# Extract text chunks from the payload
|
| 11 |
+
chunks = payload.get("inputs", [])
|
|
|
|
|
|
|
| 12 |
|
|
|
|
|
|
|
| 13 |
# Generate embeddings
|
| 14 |
embeddings = self.embedding_model.encode(chunks, convert_to_tensor=True)
|
| 15 |
|