Update app.py

app.py

@@ -55,7 +55,7 @@ def rank_images(model_name, images, text):
         results = model_pipe(images=images, texts=[text]).cpu()[:, 0].tolist()  # Perform the model inference on all images
         print("Initial results: should be imgs x texts", results)
         ranked_results = sorted(zip(images, results), key=lambda x: x[1], reverse=True)  # Rank results
-        ranked_images = [img for img, score in ranked_results]
+        ranked_images = [(img, f"Rank: {rank + 1} - Score: {score:.2f}") for rank, (img, score) in enumerate(ranked_results)]  # Pair images with their scores and rank
         print("Ranked Results:", ranked_results)
     except RuntimeError as e:
         print(f"RuntimeError during model inference: {e}")
@@ -86,7 +86,8 @@ demo_vqascore_ranking = gr.Interface(
     ],  # define the types of inputs
     outputs=gr.Gallery(label="Ranked Images"),  # 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."
+    description="This model ranks a gallery of images based on their similarity to a text prompt.",
+    allow_flagging='never'  
 )
 
 # Combine the demos into a tabbed interface