Wplotnikow commited on
Commit
9063ba8
·
verified ·
1 Parent(s): 1ea88b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -23,7 +23,7 @@ def create_vector_db_if_not_exists():
23
  print("База знаний не найдена. Запускаю процесс создания...")
24
  if not os.path.exists(SOURCE_DOCUMENT):
25
  raise FileNotFoundError(f"Ошибка: Файл '{SOURCE_DOCUMENT}' не найден.")
26
-
27
  loader = UnstructuredWordDocumentLoader(SOURCE_DOCUMENT)
28
  documents = loader.load()
29
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
@@ -41,7 +41,8 @@ create_vector_db_if_not_exists()
41
  embeddings = HuggingFaceEmbeddings(model_name=EMBEDDING_MODEL)
42
  db = FAISS.load_local(FAISS_INDEX_PATH, embeddings, allow_dangerous_deserialization=True)
43
  llm = HuggingFaceHub(repo_id=LLM_REPO_ID, model_kwargs={"temperature": 0.1, "max_new_tokens": 1024})
44
- qa_chain = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_ retriever(search_kwargs={"k": 3}), return_source_documents=True)
 
45
 
46
  # --- ФУНКЦИЯ ДЛЯ ИНТЕРФЕЙСА ---
47
  def get_response(message, history):
@@ -63,4 +64,4 @@ demo = gr.ChatInterface(
63
  )
64
 
65
  if __name__ == "__main__":
66
- demo.launch()
 
23
  print("База знаний не найдена. Запускаю процесс создания...")
24
  if not os.path.exists(SOURCE_DOCUMENT):
25
  raise FileNotFoundError(f"Ошибка: Файл '{SOURCE_DOCUMENT}' не найден.")
26
+
27
  loader = UnstructuredWordDocumentLoader(SOURCE_DOCUMENT)
28
  documents = loader.load()
29
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
 
41
  embeddings = HuggingFaceEmbeddings(model_name=EMBEDDING_MODEL)
42
  db = FAISS.load_local(FAISS_INDEX_PATH, embeddings, allow_dangerous_deserialization=True)
43
  llm = HuggingFaceHub(repo_id=LLM_REPO_ID, model_kwargs={"temperature": 0.1, "max_new_tokens": 1024})
44
+ # ИСПРАВЛЕННАЯ СТРОКА НИЖЕ (убран пробел в as_retriever)
45
+ qa_chain = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=db.as_retriever(search_kwargs={"k": 3}), return_source_documents=True)
46
 
47
  # --- ФУНКЦИЯ ДЛЯ ИНТЕРФЕЙСА ---
48
  def get_response(message, history):
 
64
  )
65
 
66
  if __name__ == "__main__":
67
+ demo.launch()