Spaces:
Runtime error
Runtime error
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(
|