ArturG9 commited on
Commit
dedb405
1 Parent(s): fccb1cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -45,15 +45,15 @@ def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='m
45
 
46
 
47
  # Check if vectorstore exists
48
- if os.path.exists(vectorstore_path) and os.listdir(vectorstore_path):
49
- # Load the existing vectorstore
50
- vectorstore = Chroma(persist_directory=vectorstore_path,embedding_function=embeddings)
51
- else:
52
  # Load documents from the specified data path
53
- loader = DirectoryLoader('./data/', glob="./*.txt", loader_cls=TextLoader)
54
- docs = loader.load()
55
- text_splitter = RecursiveCharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
56
- split_docs = text_splitter.split_documents(docs)
57
 
58
 
59
 
@@ -133,7 +133,7 @@ def handle_userinput(user_question,retriever):
133
 
134
  def create_conversational_rag_chain(retriever):
135
 
136
- model_path = ('qwen2-0_5b-instruct-q4_0.gguf')
137
 
138
  callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
139
 
 
45
 
46
 
47
  # Check if vectorstore exists
48
+ #if os.path.exists(vectorstore_path) and os.listdir(vectorstore_path):
49
+ # # Load the existing vectorstore
50
+ # vectorstore = Chroma(persist_directory=vectorstore_path,embedding_function=embeddings)
51
+ #else:
52
  # Load documents from the specified data path
53
+ loader = DirectoryLoader('./data/', glob="./*.txt", loader_cls=TextLoader)
54
+ docs = loader.load()
55
+ text_splitter = RecursiveCharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
56
+ split_docs = text_splitter.split_documents(docs)
57
 
58
 
59
 
 
133
 
134
  def create_conversational_rag_chain(retriever):
135
 
136
+
137
 
138
  callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
139