Update app.py

app.py

@@ -99,4 +99,16 @@ with gr.Blocks() as demo:
 
     build_btn.click(build_index, inputs=n, outputs=status)
 
+    gr.Markdown("Search the index with text or an example image")
+
+    q_text = gr.Textbox(label="Text query", placeholder="e.g., spiral galaxy")
+    q_img  = gr.Image(label="Image query", type="pil")
+    k      = gr.Slider(1, 12, value=6, step=1, label="Top K")
+
+    search_btn = gr.Button("Search")
+    gallery    = gr.Gallery(label="Results", columns=6, height=300)
+    info2      = gr.Textbox(label="Search status", lines=1)
+
+    search_btn.click(search, inputs=[q_text, q_img, k], outputs=[gallery, info2])
+
 demo.launch()