Update app.py

app.py

@@ -11,6 +11,8 @@ tqdm.pandas()
 
 #######################################################################################
 
+print("Downloading repo")
+
 # Setup transaction details
 repo_id = "bluuebunny/arxiv_abstract_embedding_mxbai_large_v1_milvus"
 repo_type = "dataset"
@@ -24,7 +26,7 @@ snapshot_download(repo_id=repo_id, repo_type=repo_type, local_dir=local_dir)
 
 # Function to convert dense vector to binary vector
 def dense_to_binary(dense_vector):
-   return np.packbits(np.where(dense_vector >= 0, 1, 0)).tobytes()
+    return np.packbits(np.where(dense_vector >= 0, 1, 0))
 
 
 # Gather fp32 files
@@ -80,7 +82,7 @@ print("Upload complete")
 
 # Function to convert dense vector to binary vector
 def binary_to_mrl(binary_vector, size=512):
-   return np.packbits(np.unpackbits(binary_vector)[:size]).tobytes()
+    return np.packbits(np.unpackbits(binary_vector)[:size])
 
 # Gather fp32 files
 binaries = glob('binary_embeddings/*.parquet')