zrtgb

app.py

@@ -211,11 +211,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=[output_gallery])
+    search_button.click(search, inputs=[query, embeds, imgs, k], outputs=[img_chunk])
 
     answer_button = gr.Button("Answer", variant="primary")
     output = gr.Textbox(label="Output")
-    answer_button.click(model_inference, inputs=[output_gallery, query], outputs=output)
+    answer_button.click(model_inference, inputs=[img_chunk, query], outputs=output)
 
 if __name__ == "__main__":
     demo.queue(max_size=10).launch(debug=True)