Update app.py

app.py

@@ -113,28 +113,61 @@ example_prompt1 = "Two dogs of the same breed playing on the grass"
 def load_example(model_name, images, prompt):
     return model_name, images, prompt
 
-# 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.")
-    model_dropdown = gr.Dropdown(["clip-flant5-xxl", "clip-flant5-xl"], value="clip-flant5-xxl",  label="Model Name")
-    gallery = gr.Gallery(label="Generated Images", elem_id="input-gallery", columns=4, allow_preview=True)
-    prompt = gr.Textbox(label="Prompt")
-    rank_button = gr.Button("Rank Images")
-    ranked_gallery = gr.Gallery(label="Ranked Images with Scores", elem_id="ranked-gallery", columns=4, allow_preview=True)
+
+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=[
+#         gr.Dropdown(["clip-flant5-xxl", "clip-flant5-xl", ], label="Model Name"), 
+#         gr.Image(type="filepath"), 
+#         gr.Textbox(label="Prompt")
+#     ],  # define the types of inputs
+#     examples=[
+#         ["clip-flant5-xl", example_imgs[0], example_prompt0],
+#         ["clip-flant5-xl", example_imgs[0], example_prompt1],
+#         ],
+#     outputs="number",  # define the type of output
+#     title="VQAScore",  # title of the app
+#     description="This model evaluates the similarity between an image and a text prompt."
+# )
+
+# # 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.")
+#     model_dropdown = gr.Dropdown(["clip-flant5-xxl", "clip-flant5-xl"], value="clip-flant5-xxl",  label="Model Name")
+#     gallery = gr.Gallery(label="Generated Images", elem_id="input-gallery", columns=4, allow_preview=True)
+#     prompt = gr.Textbox(label="Prompt")
+#     rank_button = gr.Button("Rank Images")
+#     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)
+#     rank_button.click(fn=rank_images, inputs=[model_dropdown, gallery, prompt], outputs=ranked_gallery)
     
-    # Custom example buttons
-    example1_button = gr.Button("Load Example 1")
-    example2_button = gr.Button("Load Example 2")
+#     # Custom example buttons
+#     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])
+#     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])
 
-    # Layout to allow user to input their own data
-    with gr.Row():
-        gr.Column([model_dropdown, gallery, prompt, rank_button])
-        gr.Column([example1_button, example2_button])
+#     # Layout to allow user to input their own data
+#     with gr.Row():
+#         gr.Column([model_dropdown, gallery, prompt, rank_button])
+#         gr.Column([example1_button, example2_button])
 
 # Launch the interface
 demo_vqascore_ranking.queue()