Improved example [skip ci]

Author: ankane

examples/pytorch_image_search.py

@@ -41,13 +41,15 @@ def generate_embeddings(inputs):
     conn.execute('DROP TABLE IF EXISTS image')
     conn.execute('CREATE TABLE image (id bigserial PRIMARY KEY, embedding vector(512))')
 
+    print('Generating embeddings')
     for data in tqdm(dataloader):
         embeddings = generate_embeddings(data[0])
 
         sql = 'INSERT INTO image (embedding) VALUES ' + ','.join(['(%s)' for _ in embeddings])
         params = [embedding for embedding in embeddings]
         conn.execute(sql, params)
 
+    print('Creating index')
     conn.execute('CREATE INDEX ON image USING hnsw (embedding vector_cosine_ops)')