mery22 commited on
Commit
711c657
1 Parent(s): d2f6d45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,7 +49,7 @@ Answer in french only
49
  repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
50
 
51
  mistral_llm = HuggingFaceEndpoint(
52
- repo_id=repo_id, max_length=2048, temperature=0.05, huggingfacehub_api_token=st.secrets["HF_TOKEN"]
53
  )
54
 
55
  # Create prompt from prompt template
@@ -62,7 +62,7 @@ prompt = PromptTemplate(
62
  llm_chain = LLMChain(llm=mistral_llm, prompt=prompt)
63
 
64
 
65
- retriever.search_kwargs = {'k':1}
66
  qa = RetrievalQA.from_chain_type(
67
  llm=mistral_llm,
68
  chain_type="stuff",
 
49
  repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
50
 
51
  mistral_llm = HuggingFaceEndpoint(
52
+ repo_id=repo_id, max_length=2048, temperature=0.02, huggingfacehub_api_token=st.secrets["HF_TOKEN"]
53
  )
54
 
55
  # Create prompt from prompt template
 
62
  llm_chain = LLMChain(llm=mistral_llm, prompt=prompt)
63
 
64
 
65
+ retriever.search_kwargs = {'k':4}
66
  qa = RetrievalQA.from_chain_type(
67
  llm=mistral_llm,
68
  chain_type="stuff",