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