Update app.py

app.py

@@ -51,7 +51,8 @@ def rank_images(model_name, images, text):
     print("Using model:", model_name)
     
     try:
-        results = model_pipe(images=images, texts=[text] * len(images)).cpu()[:, 0].tolist()  # Perform the model inference on all images
+        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]
         print("Ranked Results:", ranked_results)