update np types

src/diary_cover_similarity.py

@@ -65,14 +65,10 @@ def save_faiss_index(load_file, save_path):
             ids.append(tmp[0])
             features.append(np.array(json.loads(tmp[1])))
 
-        ids_np = np.array(ids).astype("float32")
-        features_np = np.array(features).astype("int")
-        print(ids_np)
-        print(ids.dtype)
-        print(features_np)
-        print(features_np.dtype)
-        index = faiss.IndexHNSW(128, 32)
-        print(index.is_trained)
+        ids_np = np.array(ids).astype("int")
+        features_np = np.array(features).astype("float32")
+        index = faiss.IndexHNSWFlat(128, 32)
+        print("trained: " + index.is_trained)
         index2 = faiss.IndexIDMap(index)
         index2.add_with_ids(features_np, ids_np)
         faiss.write_index(index2, save_path)