saumitras commited on
Commit
beaf3b0
Β·
unverified Β·
1 Parent(s): b3942f2

adding rag

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -108,8 +108,8 @@ def create_ui():
108
  label="Number of results"
109
  )
110
  search_btn = gr.Button("Search")
111
- status_message = gr.Textbox(label="Status")
112
- results = gr.Image(label="Retrieved Documents")
113
 
114
  # Event handlers
115
  file_input.change(
@@ -121,7 +121,7 @@ def create_ui():
121
  search_btn.click(
122
  fn=app.search_documents,
123
  inputs=[state, query_input, num_results],
124
- outputs=[results, status_message]
125
  )
126
 
127
  return demo
 
108
  label="Number of results"
109
  )
110
  search_btn = gr.Button("Search")
111
+ llm_answer = gr.Textbox(label="Answer", interactive=False)
112
+ images = gr.Image(label="Pages used for RAG")
113
 
114
  # Event handlers
115
  file_input.change(
 
121
  search_btn.click(
122
  fn=app.search_documents,
123
  inputs=[state, query_input, num_results],
124
+ outputs=[images, llm_answer]
125
  )
126
 
127
  return demo