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 = """
|