Update app.py

app.py

@@ -228,6 +228,8 @@ print(f"Shape of objection_embeddings: {objection_embeddings.shape}")
 
 # Assuming you know the expected dimension of the embeddings (e.g., 768)
 expected_dim = 768  # Example value for sentence embeddings (replace with the actual dimension)
+if len(objection_embeddings.shape) == 1:
+    objection_embeddings = objection_embeddings.reshape(-1, 1)
 
 # Check if the embeddings dimensionality matches the expected dimension
 if objection_embeddings.shape[1] != expected_dim: