diff options
Diffstat (limited to 'rag/ui.py')
-rw-r--r-- | rag/ui.py | 6 |
1 files changed, 2 insertions, 4 deletions
@@ -1,13 +1,11 @@ import streamlit as st - +from dotenv import load_dotenv from langchain_community.document_loaders.blob_loaders import Blob -from dotenv import load_dotenv -from rag.generator import get_generator, MODELS +from rag.generator import MODELS, get_generator from rag.generator.prompt import Prompt from rag.retriever.retriever import Retriever - if __name__ == "__main__": load_dotenv() retriever = Retriever() |