SeanLee97 commited on
Commit
b5b1d61
·
1 Parent(s): 5a30220

deduplicate

Browse files
Files changed (1) hide show
  1. app.py +6 -4
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
- image = ds[image_idx]['image']
126
- images.append(image)
 
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(