Spaces:
Runtime error
Runtime error
File size: 343 Bytes
ec96972 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
import faiss
from sentence_transformers import SentenceTransformer
import numpy as np
model = SentenceTransformer("all-MiniLM-L6-v2")
def build_faiss_index(chunks):
embeddings = model.encode(chunks)
dimension = embeddings.shape[1]
index = faiss.IndexFlatL2(dimension)
index.add(np.array(embeddings))
return index, chunks
|