Update app.py

app.py

@@ -19,9 +19,9 @@ data = dataset["train"]
 example_embedding = data[0]['embedding']
 print(f"Example embedding shape: {np.array(example_embedding).shape}")
 
-# Ensure embeddings are 2-dimensional and of type float16
+# Ensure embeddings are 2-dimensional and of type string
 def ensure_2d_embeddings(embeddings):
-    embeddings = np.array(embeddings, dtype=np.float16)
+    embeddings = np.array(embeddings, dtype=np.string)
     if embeddings.ndim == 1:
         embeddings = embeddings.reshape(1, -1)
     return embeddings