k3ybladewielder commited on
Commit
f819bed
·
verified ·
1 Parent(s): 73bbdbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -23,7 +23,10 @@ warnings.filterwarnings("ignore")
23
  from langchain_community.document_loaders import RecursiveUrlLoader
24
  import yaml
25
 
26
-
 
 
 
27
 
28
  # ------------ criando vs -----------------
29
 
@@ -69,11 +72,6 @@ os.makedirs(VS_BASE, exist_ok=True)
69
  vector_store.save_local(VS_BASE)
70
  print(f"vs_base salva em {VS_BASE}")
71
 
72
- # --- CONFIGURAÇÕES DE MODELOS ---
73
- LLM_MODEL = 'google/gemma-3-4b-it'
74
- # LLM_MODEL = 'google/gemma-3-1b-it'
75
- EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
76
-
77
  # --- CONFIGURAÇÃO DO TOKEN HF ---
78
  try:
79
  with open("../config.yaml", "r") as file:
 
23
  from langchain_community.document_loaders import RecursiveUrlLoader
24
  import yaml
25
 
26
+ # --- CONFIGURAÇÕES DE MODELOS ---
27
+ LLM_MODEL = 'google/gemma-3-4b-it'
28
+ # LLM_MODEL = 'google/gemma-3-1b-it'
29
+ EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
30
 
31
  # ------------ criando vs -----------------
32
 
 
72
  vector_store.save_local(VS_BASE)
73
  print(f"vs_base salva em {VS_BASE}")
74
 
 
 
 
 
 
75
  # --- CONFIGURAÇÃO DO TOKEN HF ---
76
  try:
77
  with open("../config.yaml", "r") as file: