fdf

app.py

@@ -223,7 +223,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             img_chunk = gr.State(value=[])
 
         with gr.Column(scale=3):
-            gr.Markdown("## 2️⃣ Search")
+            gr.Markdown("## 2️⃣ Search with ColPali")
             query = gr.Textbox(placeholder="Enter your query here", label="Query")
             k = gr.Slider(minimum=1, maximum=10, step=1, label="Number of results", value=5)
             search_button = gr.Button("🔍 Search", variant="primary")
@@ -241,6 +241,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     convert_button.click(index, inputs=[file, embeds], outputs=[message, embeds, imgs])
     search_button.click(search, inputs=[query, embeds, imgs, k], outputs=[output_gallery])
 
+    gr.Markdown("## 3️⃣ Get your answer with Idefics")
     answer_button = gr.Button("Answer", variant="primary")
     output = gr.Markdown(label="Output")
     answer_button.click(model_inference, inputs=[output_gallery, query], outputs=output)