k3ybladewielder commited on
Commit
7243c9b
·
verified ·
1 Parent(s): 4d5bf0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -30,7 +30,8 @@ os.makedirs(LOCAL_FOLDER, exist_ok=True)
30
  os.makedirs(VS_BASE, exist_ok=True)
31
 
32
  # --- CONFIGURAÇÕES DE MODELOS ---
33
- LLM_MODEL = 'google/gemma-3-4b-it'
 
34
  EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
35
 
36
  # ------------ criando vs -----------------
@@ -109,10 +110,10 @@ except Exception as e:
109
  # --- 4. Retriever a partir da vector store (usando similarity como exemplo) ---
110
  retriever = faiss_store.as_retriever(search_type="similarity", search_kwargs={"k": 10})
111
 
112
- retriever = faiss_store.as_retriever(
113
- search_type="mmr",
114
- search_kwargs={"k": 5, "fetch_k": 20, "lambda_mult": 0.7}
115
- )
116
 
117
  # --- 5. PromptTemplate personalizado ---
118
  custom_prompt_template = """
 
30
  os.makedirs(VS_BASE, exist_ok=True)
31
 
32
  # --- CONFIGURAÇÕES DE MODELOS ---
33
+ # LLM_MODEL = 'google/gemma-3-4b-it'
34
+ LLM_MODEL = 'google/gemma-3-12b-it'
35
  EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
36
 
37
  # ------------ criando vs -----------------
 
110
  # --- 4. Retriever a partir da vector store (usando similarity como exemplo) ---
111
  retriever = faiss_store.as_retriever(search_type="similarity", search_kwargs={"k": 10})
112
 
113
+ # retriever = faiss_store.as_retriever(
114
+ # search_type="mmr",
115
+ # search_kwargs={"k": 5, "fetch_k": 20, "lambda_mult": 0.7}
116
+ # )
117
 
118
  # --- 5. PromptTemplate personalizado ---
119
  custom_prompt_template = """