absiitr commited on
Commit
978b7eb
·
verified ·
1 Parent(s): 15a93b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -227,7 +227,7 @@ def process_pdf(uploaded_file):
227
  )
228
 
229
  vectorstore = Chroma.from_documents(chunks, embeddings)
230
- retriever = vectorstore.as_retriever(search_kwargs={"k": 4})
231
 
232
  st.session_state.vectorstore = vectorstore
233
  st.session_state.retriever = retriever
@@ -265,7 +265,7 @@ Answer on your behalf, don't say based on the context...
265
  {"role": "system", "content": "Use only the PDF content."},
266
  {"role": "user", "content": prompt}
267
  ],
268
- temperature=0.2
269
  )
270
  return response.choices[0].message.content.strip(), len(docs), None
271
  except Exception as e:
 
227
  )
228
 
229
  vectorstore = Chroma.from_documents(chunks, embeddings)
230
+ retriever = vectorstore.as_retriever(search_kwargs={"k": 5})
231
 
232
  st.session_state.vectorstore = vectorstore
233
  st.session_state.retriever = retriever
 
265
  {"role": "system", "content": "Use only the PDF content."},
266
  {"role": "user", "content": prompt}
267
  ],
268
+ temperature=0.1
269
  )
270
  return response.choices[0].message.content.strip(), len(docs), None
271
  except Exception as e: