Spaces:
Running
Running
deduplicate
Browse files
app.py
CHANGED
@@ -120,13 +120,15 @@ if __name__ == "__main__":
|
|
120 |
images = []
|
121 |
captions = []
|
122 |
similarities = []
|
123 |
-
|
|
|
124 |
for similarity, caption, image_idx in results:
|
125 |
-
|
126 |
-
|
|
|
127 |
captions.append(caption)
|
128 |
similarities.append(f"{similarity:.4f}")
|
129 |
-
|
130 |
return images, captions, similarities
|
131 |
|
132 |
demo = gr.Interface(
|
|
|
120 |
images = []
|
121 |
captions = []
|
122 |
similarities = []
|
123 |
+
|
124 |
+
visited_images = set()
|
125 |
for similarity, caption, image_idx in results:
|
126 |
+
if image_idx not in visited_images:
|
127 |
+
visited_images.add(image_idx)
|
128 |
+
images.append(ds[image_idx]['image'])
|
129 |
captions.append(caption)
|
130 |
similarities.append(f"{similarity:.4f}")
|
131 |
+
|
132 |
return images, captions, similarities
|
133 |
|
134 |
demo = gr.Interface(
|