Update UI

app.py

@@ -173,9 +173,9 @@ css = """
 file = gr.File(file_types=["pdf"], file_count="multiple", label="pdfs")
 query = gr.Textbox(placeholder="Enter your query here", label="query")
 
-with gr.Blocks(title="ColPali + Pixtral", theme=gr.themes.Soft(), css=css) as demo:
+with gr.Blocks(title="Question Answering with ColPali & Pixtral", theme=gr.themes.Soft(), css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("# ColPali + Pixtral")
+        gr.Markdown("# Question Answering with ColPali & Pixtral")
 
         with gr.Row():
             gr.Examples(
@@ -185,11 +185,10 @@ with gr.Blocks(title="ColPali + Pixtral", theme=gr.themes.Soft(), css=css) as de
 
         with gr.Row():
             with gr.Column(scale=2):
-                gr.Markdown("## Upload PDFs")
-
+                gr.Markdown("## Index PDFs")
                 file.render()
+                convert_button = gr.Button("🔄 Run", variant="primary")
                 message = gr.Textbox("Files not yet uploaded", label="Status")
-                convert_button = gr.Button("🔄 Index documents")
                 embeds = gr.State(value=[])
                 imgs = gr.State(value=[])
                 img_chunk = gr.State(value=[])
@@ -200,7 +199,7 @@ with gr.Blocks(title="ColPali + Pixtral", theme=gr.themes.Soft(), css=css) as de
                 k = gr.Slider(
                     minimum=1, maximum=4, step=1, label="Number of results", value=1
                 )
-                search_button = gr.Button("🔍 Search", variant="primary")
+                search_button = gr.Button("🔍 Run", variant="primary")
 
         # Define the actions
 
@@ -216,7 +215,7 @@ with gr.Blocks(title="ColPali + Pixtral", theme=gr.themes.Soft(), css=css) as de
         )
 
         gr.Markdown("## Get your answer with Pixtral")
-        answer_button = gr.Button("Answer", variant="primary")
+        answer_button = gr.Button("Run", variant="primary")
         output = gr.Markdown(label="Output")
         answer_button.click(
             model_inference, inputs=[output_gallery, query], outputs=output