tidying

app.py

@@ -54,12 +54,19 @@ embeddings /= np.linalg.norm(embeddings, axis=-1)[:, None]
 
 n, d = embeddings.shape
 
-print("annoy indexing")
-annoy_index = annoy.AnnoyIndex(d, 'angular')
-for i, vec in enumerate(embeddings):
-    annoy_index.add_item(i, vec)
-annoy_index.build(10)
-print("done")
+
+def build_ann_index(embeddings):
+    print("annoy indexing")
+    n, d = embeddings.shape
+    annoy_index = annoy.AnnoyIndex(d, "angular")
+    for i, vec in enumerate(embeddings):
+        annoy_index.add_item(i, vec)
+    annoy_index.build(10)
+    print("done")
+    return annoy_index
+
+
+annoy_index = build_ann_index(embeddings)
 
 
 filenames = data["filenames"]