Update space
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import ollama
|
| 2 |
import faiss
|
| 3 |
import gradio as gr
|
|
@@ -12,7 +13,8 @@ print("Inicializando...")
|
|
| 12 |
def inicializar_rag():
|
| 13 |
"""Inicializa o modelo de embeddings, documentos e índice FAISS a partir dos arquivos de capítulo."""
|
| 14 |
# Diretório onde os arquivos de capítulo estão localizados
|
| 15 |
-
|
|
|
|
| 16 |
|
| 17 |
# Padrão para encontrar arquivos como b1.txt, b2.txt, ..., s1.txt, ..., d1.txt, ..., e1.txt, etc.
|
| 18 |
padrao_arquivos = os.path.join(diretorio_rag, "[bsde][1-9].txt")
|
|
|
|
| 1 |
+
from pathlib import Path
|
| 2 |
import ollama
|
| 3 |
import faiss
|
| 4 |
import gradio as gr
|
|
|
|
| 13 |
def inicializar_rag():
|
| 14 |
"""Inicializa o modelo de embeddings, documentos e índice FAISS a partir dos arquivos de capítulo."""
|
| 15 |
# Diretório onde os arquivos de capítulo estão localizados
|
| 16 |
+
base_dir = Path(__file__).resolve().parent
|
| 17 |
+
diretorio_rag = base_dir / "RAG"
|
| 18 |
|
| 19 |
# Padrão para encontrar arquivos como b1.txt, b2.txt, ..., s1.txt, ..., d1.txt, ..., e1.txt, etc.
|
| 20 |
padrao_arquivos = os.path.join(diretorio_rag, "[bsde][1-9].txt")
|