sentence-transformers How to use chenbowen184/instacart-two-tower-sbert with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("chenbowen184/instacart-two-tower-sbert")
sentences = [
"[+1d w0h15] Butter, Garlic, Applewood Smoked Center Cut Uncured Bacon, Whole Milk, Pie Pans, Large, Mint Chip Ice Cream, Rocky Road Ice Cream; [+6d w6h15] Paste, Red Curry, Coconut Milk, Italian Kitchen Red Wine Vinegar, Jack Habanero Cheese, Garlic, Sourdough Baguette, Fresh Ginger Root, Dry Roasted Lightly Salted Peanuts, Limes, 100% Pure Sesame Seed Oil, Organic Shiitake Mushrooms, Unsalted Chicken Cooking Stock, Broccoli Crown. Next: +13d w6h10",
"Product: Banana. Aisle: fresh fruits. Department: produce.",
"Product: Whole Chicken. Aisle: poultry counter. Department: meat seafood.",
"Product: Organic Strawberries. Aisle: fresh fruits. Department: produce."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
