ashok2216 commited on
Commit
d60eba5
·
verified ·
1 Parent(s): 97ac7bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -56,12 +56,16 @@ def query_pdf_data(collection, query, retriever_model):
56
  query_texts=[query],
57
  n_results=3
58
  )
 
59
  context = " ".join([doc for doc in results["documents"][0]])
60
  answer = retriever_model(f"Context: {context}\nQuestion: {query}")
61
  return answer, results["metadatas"]
62
 
63
  # Streamlit Interface
64
  def main():
 
 
 
65
  st.title("PDF Chatbot with Retrieval-Augmented Generation")
66
  st.write("Upload a PDF, and ask questions about its content!")
67
 
 
56
  query_texts=[query],
57
  n_results=3
58
  )
59
+
60
  context = " ".join([doc for doc in results["documents"][0]])
61
  answer = retriever_model(f"Context: {context}\nQuestion: {query}")
62
  return answer, results["metadatas"]
63
 
64
  # Streamlit Interface
65
  def main():
66
+ image = Image.open('LOGO.PNG')
67
+ st.image(
68
+ image, width=250)
69
  st.title("PDF Chatbot with Retrieval-Augmented Generation")
70
  st.write("Upload a PDF, and ask questions about its content!")
71