Update app.py

app.py

@@ -32,6 +32,8 @@ def set_labels(text):
 get_caption = gr.load("ryaalbr/caption", src="spaces", hf_token=environ["api_key"])
 def generate_text(image, model_name):
     return get_caption(image, model_name)
+
+get_images = gr.load("ryaalbr/ImageSearch", src="spaces", hf_token=environ["api_key"])
     
 with gr.Blocks() as demo:
 
@@ -76,7 +78,14 @@ with gr.Blocks() as demo:
             caption = gr.Textbox(label='Caption')
         get_btn_cap.click(fn=rand_image, outputs=im_cap)
         #im_cap.change(generate_text, inputs=im_cap, outputs=caption)
-        caption_btn.click(generate_text, inputs=[im_cap, model_name], outputs=caption)        
+        caption_btn.click(generate_text, inputs=[im_cap, model_name], outputs=caption)
+
+    with gr.Tab("Image Search"):
+        with gr.Column(variant="panel"):
+            desc = gr.Textbox(show_label=False, placeholder="Enter description").style(container=False)
+            search_btn = gr.Button("Find Images").style(full_width=False)
+        gallery = gr.Gallery(show_label=False).style(grid=(2,2,3,5))
+        search_btn.click(get_images,inputs=desc, outputs=gallery)
 
 demo.queue()
 demo.launch()