Stéphanie Kamgnia Wonkap commited on
Commit
12e199b
1 Parent(s): 3504448

fixing indention in app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -71,9 +71,10 @@ def main():
71
  st.session_state.embedding_model)
72
  if (user_query) and (st.button("Get Answer")):
73
  num_doc_before_rerank=15
74
- st.session_state.retriever= st.session_state.KNOWLEDGE_VECTOR_DATABASE.as_retriever(search_type="similarity",
 
75
 
76
- st.write("### Please wait while we are getting the answer.....") search_kwargs={"k": num_doc_before_rerank})
77
  llm = ChatNVIDIA(
78
  model=READER_MODEL_NAME,
79
  api_key= os.get("NVIDIA_API_KEY"),
 
71
  st.session_state.embedding_model)
72
  if (user_query) and (st.button("Get Answer")):
73
  num_doc_before_rerank=15
74
+ st.session_state.retriever= st.session_state.KNOWLEDGE_VECTOR_DATABASE.as_retriever(search_type="similarity",
75
+ search_kwargs={"k": num_doc_before_rerank})
76
 
77
+ st.write("### Please wait while we are getting the answer.....")
78
  llm = ChatNVIDIA(
79
  model=READER_MODEL_NAME,
80
  api_key= os.get("NVIDIA_API_KEY"),