Update transformers_similarity.py (#6280)

Fixing the Document examples
This commit is contained in:
Tuana Çelik 2023-11-13 09:35:00 +01:00 committed by GitHub
parent 92a8704de4
commit bf637e9c7e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -25,7 +25,7 @@ class TransformersSimilarityRanker:
from haystack.preview.components.rankers import TransformersSimilarityRanker
ranker = TransformersSimilarityRanker()
docs = [Document(text="Paris"), Document(text="Berlin")]
docs = [Document(content="Paris"), Document(content="Berlin")]
query = "City in Germany"
output = ranker.run(query=query, documents=docs)
docs = output["documents"]