Update app.py

app.py

@@ -19,8 +19,6 @@ import json
 cache_dir = '/data/KB'
 os.makedirs(cache_dir, exist_ok=True)
 
-
-
 @spaces.GPU
 def weighted_mean_pooling(hidden, attention_mask):
     attention_mask_ = attention_mask * attention_mask.cumsum(dim=1)
@@ -274,7 +272,16 @@ with gr.Blocks() as app:
         query_input = gr.Text(label="Your Queston")
         topk_input = inputs=gr.Number(value=5, minimum=1, maximum=10, step=1, label="Number of pages to retrieve")
         retrieve_button = gr.Button("Step2: Retrieve Pages")
-    
+
+    with gr.Row():
+        gr.Examples(
+            examples=[
+                ["main_figure.pdf", "What is RAG-V?"],
+                ["main_figure.pdf", "How does RAG-V perform?"]
+            ],
+            inputs=[file_input, query_input],
+        )
+
     with gr.Row():
         images_output = gr.Gallery(label="Retrieved Pages")