Yoxas commited on
Commit
200204a
·
verified ·
1 Parent(s): 669bc11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ print(f"Example embedding shape: {np.array(example_embedding).shape}")
21
 
22
  # Ensure embeddings are 2-dimensional and of type string
23
  def ensure_2d_embeddings(embeddings):
24
- embeddings = np.array(embeddings, dtype=np.float64)
25
  if embeddings.ndim == 1:
26
  embeddings = embeddings.reshape(1, -1)
27
  return embeddings
 
21
 
22
  # Ensure embeddings are 2-dimensional and of type string
23
  def ensure_2d_embeddings(embeddings):
24
+ embeddings = np.array(embeddings, dtype=np.float32)
25
  if embeddings.ndim == 1:
26
  embeddings = embeddings.reshape(1, -1)
27
  return embeddings