vzefv

app.py

@@ -232,11 +232,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
  output_gallery = gr.Gallery(label="Retrieved Documents", height=600, show_label=True)
 
  convert_button.click(index, inputs=[file, embeds], outputs=[message, embeds, imgs])
- search_button.click(search, inputs=[query, embeds, imgs, k], outputs=[img_chunk])
+ search_button.click(search, inputs=[query, embeds, imgs, k], outputs=[output_gallery])
 
  answer_button = gr.Button("Answer", variant="primary")
  output = gr.Textbox(label="Output")
- answer_button.click(model_inference, inputs=[img_chunk, query], outputs=output)
+ answer_button.click(model_inference, inputs=[output_gallery, query], outputs=output)
 
 if __name__ == "__main__":
  demo.queue(max_size=10).launch(debug=True)