Update app.py

app.py

@@ -40,23 +40,23 @@ def lexica(prompt, limit_size = 128, ratio_size = 256 + 128):
     lex = list(map(lambda x: min_dim_to_size(x, ratio_size)[1], lex))
     return lex
 
-with gr.Blocks() as demo:
+with gr.Blocks("custom.css") as demo:
     with gr.Row():
         with gr.Tabs():
             with gr.Row():
                 inputs = gr.Textbox(label = 'Enter prompt to search Lexica.art')
                 #gr.Slider(label='Number of images ', minimum = 4, maximum = 20, step = 1, value = 4)]
-                text_button = gr.Button("Retrieve Images").style(full_width=False)
-            i = gr.Image()
+                text_button = gr.Button("Retrieve Images", elem_id="run_button")
+            i = gr.Image(elem_id="result-image")
         with gr.Tabs():
             #### Try to click on a image in the gallery, and the gallery order will changed
             with gr.Column():
                 title = gr.Markdown(
                     value="### Click on a Image in the gallery to select it, and the grid order will changed",
                     visible=True,
-                    elem_id="selected_image",
+                    elem_id="selected_model",
                 )
-                outputs = gr.Gallery(lable='Output gallery').style(grid=3,height=768,
+                outputs = gr.Gallery(lable='Output gallery', elem_id="gallery",).style(grid=3,height=768,
                     allow_preview=False, label = "retrieve Images")
                 #gr.Dataframe(label='prompts for corresponding images')]