Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
114 |
-
|
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 = """
|