Update app.py

app.py

@@ -196,7 +196,7 @@ def get_example():
         ]
 
 with gr.Blocks(theme= "gstaff/xkcd" ) as demo:
-    gr.Markdown("# ColPali + Qwen2VL 2B: Efficient Document Retrieval with Vision Language Models 📚")
+    gr.Markdown("🤓 Mulimoal RAGBOT: Ask Me About the Images 🌅 in Your PDF📚")
     
 
     with gr.Row():
@@ -205,16 +205,16 @@ with gr.Blocks(theme= "gstaff/xkcd" ) as demo:
             file = gr.File(file_types=["pdf"], file_count="multiple", label="Upload PDFs")
 
             
-            message = gr.Textbox("Files not yet uploaded", label="Status")
+            message = gr.Textbox("Files not yet chunked", label="Status")
             convert_button = gr.Button("🔄 Index documents")
             embeds = gr.State(value=[])
             imgs = gr.State(value=[])
             img_chunk = gr.State(value=[])
 
         with gr.Column(scale=3):
-            gr.Markdown("## 2️⃣ Search with ColPali")
+            gr.Markdown("## 2️⃣ Search the Image")
             query = gr.Textbox(placeholder="Enter your query here", label="Query")
-            k = gr.Slider(minimum=1, maximum=10, step=1, label="Number of results", value=1)
+            k = gr.Slider(minimum=1, maximum=10, step=1, label="batch", value=1)
             search_button = gr.Button("🔍 Search", variant="primary")
     
     with gr.Row():
@@ -225,13 +225,13 @@ with gr.Blocks(theme= "gstaff/xkcd" ) as demo:
 
     # Define the actions
     
-    output_gallery = gr.Gallery(label="Retrieved Documents", height=600, show_label=True)
+    output_gallery = gr.Gallery(label="Retrieved Images From Your 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])
 
-    gr.Markdown("## 3️⃣ Get your answer with Qwen2-VL")
-    answer_button = gr.Button("Answer", variant="primary")
+    gr.Markdown("## 3️⃣ Get your answer")
+    answer_button = gr.Button("💁‍♀️Answer", variant="primary")
     output = gr.Markdown(label="Output")
     answer_button.click(model_inference, inputs=[output_gallery, query], outputs=output)