Nielo47 commited on
Commit
ba063d8
·
1 Parent(s): 15cb671

Update space

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
- diretorio_rag = r".\RAG"
 
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")