Views
No views yet
1from sentence_transformers import SentenceTransformer
2
3# Download from the 🤗 Hub
4model = SentenceTransformer("embeddinggemma-pms-16384")
5
6# Run inference with queries and documents
7query = "My query"
8documents = [
9 "Chunk 1",
10 "Chunk 2",
11 "Chunk 3",
12]
13query_embeddings = model.encode_query(query)
14document_embeddings = model.encode_document(documents)
15print(query_embeddings.shape, document_embeddings.shape)
16
17# Compute similarities to determine a ranking
18similarities = model.similarity(query_embeddings, document_embeddings)
19print(similarities)