From 34965e2cdb026bfaf7facf5c520848ce19120251 Mon Sep 17 00:00:00 2001 From: navanchauhan Date: Sat, 21 May 2022 11:11:59 -0600 Subject: fixed code for pushing to pinecone --- db2pc.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'db2pc.py') diff --git a/db2pc.py b/db2pc.py index b6de15a..4acddcb 100644 --- a/db2pc.py +++ b/db2pc.py @@ -22,7 +22,7 @@ batch_size = 32 df = pd.read_sql("Select * from movies", engine) df["combined_text"] = df["title"] + ": " + df["overview"].fillna('') + " - " + df["tagline"].fillna('') + " Genres:- " + df["genres"].fillna('') -print(len(df["combined_text"].tolist())) +print(f'Length of Combined Text: {len(df["combined_text"].tolist())}') for x in tqdm(range(0,len(df),batch_size)): to_send = [] @@ -31,7 +31,7 @@ for x in tqdm(range(0,len(df),batch_size)): embeddings = model.encode(sentences) for idx, value in enumerate(trakt_ids): to_send.append( - { - value: embeddings[idx].tolist() - }) - index.upsert(to_send) + ( + str(value), embeddings[idx].tolist() + )) + index.upsert(to_send) \ No newline at end of file -- cgit v1.2.3