MuntasirHossain commited on
Commit
41aff13
·
verified ·
1 Parent(s): 877b6b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -135,8 +135,8 @@ def upload_file(file_obj):
135
  def demo():
136
  # with gr.Blocks(theme=gr.themes.Default(primary_hue="sky")) as demo:
137
  with gr.Blocks(theme=gr.themes.Default(primary_hue="red", secondary_hue="pink", neutral_hue = "sky")) as demo:
138
- vector_db = gr.State(None)
139
- qa_chain = gr.State(None)
140
  gr.HTML("<center><h1>RAG PDF chatbot</h1><center>")
141
  gr.Markdown("""<b>Query your PDF documents!</b> This AI agent is designed to perform retrieval augmented generation (RAG) on PDF documents. The app is hosted on Hugging Face Hub for the sole purpose of demonstration. \
142
  <b>Please do not upload confidential documents.</b>
@@ -209,8 +209,8 @@ def demo():
209
  inputs=None, \
210
  outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
211
  queue=False)
212
- # demo.queue().launch(debug=True)
213
- demo.launch(debug=True)
214
 
215
 
216
  if __name__ == "__main__":
 
135
  def demo():
136
  # with gr.Blocks(theme=gr.themes.Default(primary_hue="sky")) as demo:
137
  with gr.Blocks(theme=gr.themes.Default(primary_hue="red", secondary_hue="pink", neutral_hue = "sky")) as demo:
138
+ vector_db = gr.State()
139
+ qa_chain = gr.State()
140
  gr.HTML("<center><h1>RAG PDF chatbot</h1><center>")
141
  gr.Markdown("""<b>Query your PDF documents!</b> This AI agent is designed to perform retrieval augmented generation (RAG) on PDF documents. The app is hosted on Hugging Face Hub for the sole purpose of demonstration. \
142
  <b>Please do not upload confidential documents.</b>
 
209
  inputs=None, \
210
  outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], \
211
  queue=False)
212
+ demo.queue().launch(debug=True)
213
+ # demo.launch(debug=True)
214
 
215
 
216
  if __name__ == "__main__":