Update app.py

app.py

@@ -132,7 +132,12 @@ def load_example(model_name, images, prompt):
 
 # Create the second demo: VQAScore Ranking
 with gr.Blocks() as demo_vqascore_ranking:
-    gr.Markdown("# VQAScore Ranking\nThis model ranks a gallery of images based on their similarity to a text prompt.")
+    # gr.Markdown("# VQAScore Ranking\nThis model ranks a gallery of images based on their similarity to a text prompt.")
+    gr.Markdown("""
+    # VQAScore Ranking
+    This model ranks a gallery of images based on their similarity to a text prompt.
+    **Please retry if the model fails to load for the first time. Once the model is loaded to GPUs, the ranking process will be extremely fast.**
+    """)
     
     with gr.Row():
         with gr.Column():
@@ -140,7 +145,6 @@ with gr.Blocks() as demo_vqascore_ranking:
             prompt = gr.Textbox(label="Prompt")
             gallery = gr.Gallery(label="Generated Images", elem_id="input-gallery", columns=4, allow_preview=True)
             rank_button = gr.Button("Rank Images")
-            rank_button.click(fn=rank_images, inputs=[model_dropdown, gallery, prompt], outputs=ranked_gallery)
             
         with gr.Column():
             example1_button = gr.Button("Load Example 1")
@@ -149,6 +153,8 @@ with gr.Blocks() as demo_vqascore_ranking:
             example2_button.click(fn=lambda: load_example("clip-flant5-xxl", example_imgs, example_prompt1), inputs=[], outputs=[model_dropdown, gallery, prompt])
     
     ranked_gallery = gr.Gallery(label="Ranked Images with Scores", elem_id="ranked-gallery", columns=4, allow_preview=True)
+
+    rank_button.click(fn=rank_images, inputs=[model_dropdown, gallery, prompt], outputs=ranked_gallery)
     
 # # Create the second demo
 # with gr.Blocks() as demo_vqascore_ranking: