diff --git a/langchain/embeddings/tensorflow_hub.py b/langchain/embeddings/tensorflow_hub.py index 25e63949c4d..0dc439233ad 100644 --- a/langchain/embeddings/tensorflow_hub.py +++ b/langchain/embeddings/tensorflow_hub.py @@ -66,5 +66,5 @@ class TensorflowHubEmbeddings(BaseModel, Embeddings): Embeddings for the text. """ text = text.replace("\n", " ") - embedding = self.embed(text).numpy()[0] + embedding = self.embed([text]).numpy()[0] return embedding.tolist()