Update app.py

app.py

@@ -36,7 +36,15 @@ embeddings = np.array(embeddings).astype(np.float32)
 faiss.normalize_L2(embeddings)
 
 dimension = embeddings.shape[1]
-index = faiss.IndexFlatIP(dimension)
+
+nlist = 150 
+
+quantizer = faiss.IndexFlatIP(dimension)
+
+
+index = faiss.IndexIVFFlat(quantizer, dimension, nlist, faiss.METRIC_L1)
+index.train(embeddings)
+# index = faiss.IndexFlatIP(dimension)
 
 index.add(embeddings)