CoolFace
Apppublic

Allens1242/DarkPatterns

sourceHugging Faceupdated 4mo agoView on Hugging Face
0likes
embed_rag.py22 linesDownload Raw Back to root
1from sentence_transformers import SentenceTransformer2import faiss3import numpy as np4import json5 6# Load corpus7with open("bait_switch_texts.json", "r") as f:8    texts = json.load(f)9 10# Load embedding model11model = SentenceTransformer('all-MiniLM-L6-v2')12embeddings = model.encode(texts, convert_to_numpy=True)13 14# Build FAISS index15index = faiss.IndexFlatL2(embeddings.shape[1])16index.add(embeddings)17 18# Save index + text19faiss.write_index(index, "bait_switch_index.faiss")20with open("bait_switch_texts.index.json", "w") as f:21    json.dump(texts, f)22