Update app.py

app.py

@@ -90,31 +90,31 @@ demo_vqascore = gr.Interface(
     description="This model evaluates the similarity between an image and a text prompt."
 )
 
-# Create the second demo
-demo_vqascore_ranking = gr.Interface(
-    fn=rank_images,  # function to call
-    inputs=[
-        gr.Dropdown(["clip-flant5-xl", "clip-flant5-xxl"], label="Model Name"), 
-        gr.Gallery(label="Generated Images"), 
-        gr.Textbox(label="Prompt")
-    ],  # define the types of inputs
-    outputs=gr.Gallery(label="Ranked Images"),  # define the type of output
-    examples=[
-        ["clip-flant5-xl", [[img, ""] for img in example_imgs], example_prompt0],
-        ["clip-flant5-xl", [[img, ""] for img in example_imgs], example_prompt1]
+# # Create the second demo
+# demo_vqascore_ranking = gr.Interface(
+#     fn=rank_images,  # function to call
+#     inputs=[
+#         gr.Dropdown(["clip-flant5-xl", "clip-flant5-xxl"], label="Model Name"), 
+#         gr.Gallery(label="Generated Images"), 
+#         gr.Textbox(label="Prompt")
+#     ],  # define the types of inputs
+#     outputs=gr.Gallery(label="Ranked Images"),  # define the type of output
+#     examples=[
+#         ["clip-flant5-xl", [[img, ""] for img in example_imgs], example_prompt0],
+#         ["clip-flant5-xl", [[img, ""] for img in example_imgs], example_prompt1]
         
-    ],
-    title="VQAScore Ranking",  # title of the app
-    description="This model ranks a gallery of images based on their similarity to a text prompt.",
-    allow_flagging='never'  
-)
+#     ],
+#     title="VQAScore Ranking",  # title of the app
+#     description="This model ranks a gallery of images based on their similarity to a text prompt.",
+#     allow_flagging='never'  
+# )
 
-# Custom component for loading examples
-def load_example(model_name, images, prompt):
-    return model_name, images, prompt
+# # Custom component for loading examples
+# def load_example(model_name, images, prompt):
+#     return model_name, images, prompt
 
 # Create the second demo
-with gr.Blocks() as demo_vqascore_ranking_beta:
+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-xl", "clip-flant5-xxl"], label="Model Name")
     gallery = gr.Gallery(label="Generated Images", elem_id="input-gallery", columns=4, allow_preview=True)
@@ -139,7 +139,7 @@ with gr.Blocks() as demo_vqascore_ranking_beta:
         
 
 # Combine the demos into a tabbed interface
-tabbed_interface = gr.TabbedInterface([demo_vqascore, demo_vqascore_ranking, demo_vqascore_ranking_beta], ["VQAScore", "VQAScore Ranking", "Ranking Beta"])
+tabbed_interface = gr.TabbedInterface([demo_vqascore, demo_vqascore_ranking], ["VQAScore", "VQAScore Ranking"])
 
 # Launch the tabbed interface
 tabbed_interface.queue()