Spaces:
Build error
Build error
Update app.py
Browse files
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 |
-
|
|
|
|
| 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()
|