Update app.py

app.py

@@ -117,10 +117,10 @@ def load_example(model_name, images, prompt):
 with gr.Blocks() as demo_vqascore_ranking_beta:
     gr.Markdown("# VQAScore Ranking\nThis model ranks a gallery of images based on their similarity to a text prompt.")
     model_dropdown = gr.Dropdown(["clip-flant5-xl", "clip-flant5-xxl"], label="Model Name")
-    gallery = gr.Gallery(label="Generated Images")
+    gallery = gr.Gallery(label="Generated Images", elem_id="input-gallery").style(grid=(2, 2), height="200px")
     prompt = gr.Textbox(label="Prompt")
     rank_button = gr.Button("Rank Images")
-    ranked_gallery = gr.Gallery(label="Ranked Images with Scores")
+    ranked_gallery = gr.Gallery(label="Ranked Images with Scores", elem_id="ranked-gallery").style(grid=(2, 2), height="200px")
     
     rank_button.click(fn=rank_images, inputs=[model_dropdown, gallery, prompt], outputs=ranked_gallery)
     
@@ -135,6 +135,7 @@ with gr.Blocks() as demo_vqascore_ranking_beta:
     with gr.Row():
         gr.Column([model_dropdown, gallery, prompt, rank_button])
         gr.Column([example1_button, example2_button])
+
         
 
 # Combine the demos into a tabbed interface