Spaces:
Sleeping
Sleeping
Update pages/2_Chatbot_AR.py
Browse files- pages/2_Chatbot_AR.py +3 -2
pages/2_Chatbot_AR.py
CHANGED
@@ -205,7 +205,8 @@ llm_ar = HuggingFaceHub(
|
|
205 |
huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API"],
|
206 |
model_kwargs={
|
207 |
"temperature": 0.5,
|
208 |
-
"max_length": 500
|
|
|
209 |
}
|
210 |
)
|
211 |
|
@@ -280,7 +281,7 @@ def main():
|
|
280 |
return
|
281 |
|
282 |
# Récupération du contexte
|
283 |
-
context_ar_list = retrieve_context_ar(st.session_state["retriever_ar"], user_query_ar, top_k=
|
284 |
|
285 |
if context_ar_list:
|
286 |
with st.spinner("Génération de la réponse..."):
|
|
|
205 |
huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API"],
|
206 |
model_kwargs={
|
207 |
"temperature": 0.5,
|
208 |
+
"max_length": 500,
|
209 |
+
"timeout": 60
|
210 |
}
|
211 |
)
|
212 |
|
|
|
281 |
return
|
282 |
|
283 |
# Récupération du contexte
|
284 |
+
context_ar_list = retrieve_context_ar(st.session_state["retriever_ar"], user_query_ar, top_k=3)
|
285 |
|
286 |
if context_ar_list:
|
287 |
with st.spinner("Génération de la réponse..."):
|