Update app.py

app.py

@@ -114,22 +114,6 @@ def load_example(model_name, images, prompt):
     return model_name, images, prompt
 
 
-demo_vqascore_ranking = gr.Interface(
-    fn=rank_images,  # function to call
-    inputs=[
-        gr.Dropdown(["clip-flant5-xxl", "clip-flant5-xl"], label="Model Name"), 
-        gr.Gallery(label="Generated Images", elem_id="input-gallery", columns=4, allow_preview=True), 
-        gr.Textbox(label="Prompt")
-    ],  # define the types of inputs
-    examples=[
-        ["clip-flant5-xxl", example_imgs, example_prompt0],
-        ["clip-flant5-xxl", example_imgs, example_prompt1],
-    ],
-    outputs=gr.Gallery(label="Ranked Images with Scores", elem_id="ranked-gallery", columns=4, allow_preview=True),  # define the type of output
-    title="VQAScore Ranking",  # title of the app
-    description="This model ranks a gallery of images based on their similarity to a text prompt."
-)
-
 # demo_vqascore = gr.Interface(
 #     fn=generate,  # function to call
 #     inputs=[
@@ -146,6 +130,26 @@ demo_vqascore_ranking = gr.Interface(
 #     description="This model evaluates the similarity between an image and a text 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.")
+    
+    with gr.Row():
+        with gr.Column():
+            model_dropdown = gr.Dropdown(["clip-flant5-xxl", "clip-flant5-xl"], value="clip-flant5-xxl", label="Model Name")
+            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")
+            example2_button = gr.Button("Load Example 2")
+            example1_button.click(fn=lambda: load_example("clip-flant5-xxl", example_imgs, example_prompt0), inputs=[], outputs=[model_dropdown, gallery, prompt])
+            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)
+    
 # # Create the second demo
 # 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.")