Update app.py
Browse files
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.
|
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':
|
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",
|