Update rag_demo/rag/reranker.py

rag_demo/rag/reranker.py

@@ -2,9 +2,9 @@ import os
 
 from huggingface_hub import InferenceClient
 
-from rag_demo.rag.base.query import Query
-from rag_demo.rag.base.template_factory import RAGStep
-from rag_demo.preprocessing.embed import EmbeddedChunk
+from base.query import Query
+from base.template_factory import RAGStep
+from .preprocessing.embed import EmbeddedChunk
 
 
 class Reranker(RAGStep):