abdelom commited on
Commit
2e8b84c
·
verified ·
1 Parent(s): 41d8c96

Update pages/2_Chatbot_AR.py

Browse files
Files changed (1) hide show
  1. 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=5)
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..."):