bluuebunny commited on
Commit
155a763
·
verified ·
1 Parent(s): a3caeed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,7 +24,7 @@ snapshot_download(repo_id=repo_id, repo_type=repo_type, local_dir=local_dir)
24
 
25
  # Function to convert dense vector to binary vector
26
  def dense_to_binary(dense_vector):
27
- return np.where(dense_vector >= 0, 1, 0)
28
 
29
 
30
  # Gather fp32 files
@@ -80,7 +80,7 @@ print("Upload complete")
80
 
81
  # Function to convert dense vector to binary vector
82
  def binary_to_mrl(binary_vector, size=512):
83
- return binary_vector[:size]
84
 
85
  # Gather fp32 files
86
  binaries = glob('binary_embeddings/*.parquet')
 
24
 
25
  # Function to convert dense vector to binary vector
26
  def dense_to_binary(dense_vector):
27
+ return np.packbits(np.where(dense_vector >= 0, 1, 0)).tobytes()
28
 
29
 
30
  # Gather fp32 files
 
80
 
81
  # Function to convert dense vector to binary vector
82
  def binary_to_mrl(binary_vector, size=512):
83
+ return np.packbits(np.unpackbits(binary_vector)[:size]).tobytes()
84
 
85
  # Gather fp32 files
86
  binaries = glob('binary_embeddings/*.parquet')