k3ybladewielder commited on
Commit
ad73f05
·
verified ·
1 Parent(s): 1bbf731

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -18
app.py CHANGED
@@ -24,9 +24,9 @@ from langchain_community.document_loaders import RecursiveUrlLoader
24
  import yaml
25
 
26
  # #----------- local paths -----------
27
- CACHE_FOLDER = "../model/"
28
- LOCAL_FOLDER = "../model/"
29
- VS_BASE = "../vector_store/vs_base"
30
 
31
  os.makedirs(CACHE_FOLDER, exist_ok=True)
32
  os.makedirs(LOCAL_FOLDER, exist_ok=True)
@@ -81,21 +81,12 @@ os.makedirs(VS_BASE, exist_ok=True)
81
  vector_store.save_local(VS_BASE)
82
  print(f"vs_base salva em {VS_BASE}")
83
 
84
- # --- CONFIGURAÇÃO DO TOKEN HF ---
85
- # try:
86
- # with open("../config.yaml", "r") as file:
87
- # config = yaml.safe_load(file)
88
- # HF_TOKEN = config.get('HF_TOKEN')
89
- # except FileNotFoundError:
90
- # print("Arquivo config.yaml não encontrado. Certifique-se de que ele existe e está no diretório correto.")
91
- # HF_TOKEN = None
92
-
93
- # if not HF_TOKEN:
94
- # print("Token HF não encontrado no config.yaml ou arquivo não existe.")
95
- # if "HF_TOKEN" not in os.environ:
96
- # import getpass
97
- # os.environ["HF_TOKEN"] = getpass.getpass("Por favor, digite seu token da API Hugging Face: ")
98
- # HF_TOKEN = os.environ["HF_TOKEN"]
99
 
100
  # --- 1. Inicializa o LLM Hugging Face ---
101
  llm = HuggingFaceEndpoint(
 
24
  import yaml
25
 
26
  # #----------- local paths -----------
27
+ CACHE_FOLDER = "./model/"
28
+ LOCAL_FOLDER = "./model/"
29
+ VS_BASE = "./vector_store/vs_base"
30
 
31
  os.makedirs(CACHE_FOLDER, exist_ok=True)
32
  os.makedirs(LOCAL_FOLDER, exist_ok=True)
 
81
  vector_store.save_local(VS_BASE)
82
  print(f"vs_base salva em {VS_BASE}")
83
 
84
+ # --- CONFIGURAÇÃO HF ---
85
+ HF_TOKEN = os.getenv("HF_TOKEN") # Use os.getenv para ler a variável de ambiente
86
+ if not HF_TOKEN:
87
+ raise ValueError("HF_TOKEN não encontrado. Por favor, defina-o nos segredos do Hugging Face Space.")
88
+
89
+ os.environ["HF_HUB_USER_AGENT"] = "CloudWalk_Chatbot"
 
 
 
 
 
 
 
 
 
90
 
91
  # --- 1. Inicializa o LLM Hugging Face ---
92
  llm = HuggingFaceEndpoint(