Update app.py

app.py

@@ -159,6 +159,9 @@ def search(prompt: str):
      FastembedTextEmbedder,
      FastembedSparseTextEmbedder
     )
+    from haystack.components.rankers import TransformersSimilarityRanker
+    from haystack.components.joiners import DocumentJoiner
+
     start_time = time.time()
     
     # Querying
@@ -169,10 +172,12 @@ def search(prompt: str):
      model="BAAI/bge-small-en-v1.5", prefix="Represent this sentence for searching relevant passages: ")
      )
     querying.add_component("retriever", QdrantHybridRetriever(document_store=document_store))
-    
+    querying.add_component("document_joiner", DocumentJoiner())
+    querying.add_component("ranker", TransformersSimilarityRanker(model="BAAI/bge-reranker-base"))
     querying.connect("sparse_text_embedder.sparse_embedding", "retriever.query_sparse_embedding")
     querying.connect("dense_text_embedder.embedding", "retriever.query_embedding")
-    
+    querying.connect("retriever", "document_joiner")
+    querying.connect("document_joiner", "ranker")
     question = "Cosa sono i marker tumorali?"
     
     results = querying.run(