Spaces:
Paused
Paused
| from sentence_transformers import SentenceTransformer | |
| import faiss, os, pickle | |
| model = SentenceTransformer("all-MiniLM-L6-v2") | |
| index = faiss.IndexFlatL2(384) | |
| docs = [] | |
| def ingest(text): | |
| emb = model.encode([text]) | |
| index.add(emb) | |
| docs.append(text) | |
| faiss.write_index(index, "rag/vectorstore/index.faiss") | |
| pickle.dump(docs, open("rag/vectorstore/docs.pkl", "wb")) | |