Fix imports

1 files changed, 2 insertions, 2 deletions

diff --git a/rag/ui.py b/rag/ui.py
index 37ab664..18e6bd1 100644
--- a/rag/ui.py
+++ b/rag/ui.py
@@ -4,8 +4,8 @@ from langchain_community.document_loaders.blob_loaders import Blob
 
 from dotenv import load_dotenv
 from rag.generator import get_generator, MODELS
-from generator.prompt import Prompt
-from retriever.retriever import Retriever
+from rag.generator.prompt import Prompt
+from rag.retriever.retriever import Retriever
 
 
 if __name__ == "__main__":