Update app.py

app.py

@@ -249,8 +249,16 @@ objection_embeddings = np.array(objection_embeddings, dtype="float32")
 print(f"Shape of objection_embeddings: {objection_embeddings.shape}")
 
 # Check if the embeddings dimensionality matches the expected dimension
-if objection_embeddings.shape[1] != expected_dim:
-    raise ValueError(f"Dimensionality of embeddings {objection_embeddings.shape[1]} does not match expected dimension {expected_dim}.")
+# If the shape is (1,), reshape it to (1, expected_dim)
+if len(objection_embeddings.shape) == 1:
+    # Reshape for a single embedding, turn it into a 2D array
+    objection_embeddings = objection_embeddings.reshape(1, -1)
+elif len(objection_embeddings.shape) == 2:
+    # Check if second dimension matches expected_dim
+    if objection_embeddings.shape[1] != expected_dim:
+        raise ValueError(f"Dimensionality of embeddings {objection_embeddings.shape[1]} does not match expected dimension {expected_dim}.")
+else:
+    raise ValueError(f"Unexpected shape for objection embeddings: {objection_embeddings.shape}")
 
 # Add the embeddings to the Faiss index
 faiss_index.add(objection_embeddings)