RobotJelly commited on
Commit
927e71f
·
1 Parent(s): d489fca
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -112,7 +112,6 @@ def image_search(search_text, search_image, option):
112
  #matched_images = find_matches(text_features, photo_features, photo_ids, 4)
113
  #similarity = util.cos_sim(image_emb, img_emb)
114
  #matched_results = util.semantic_search(image_emb, img_emb, 4)[0]
115
- PILIMAGE.fromarray(image.astype('uint8'), 'RGB')
116
  image_emb = model.encode([Image.fromarray(search_image.astype('uint8'), 'RGB')], convert_to_tensor=True)
117
  similarity = util.cos_sim(img_emb, image_emb)
118
  return display_matches(similarity)
 
112
  #matched_images = find_matches(text_features, photo_features, photo_ids, 4)
113
  #similarity = util.cos_sim(image_emb, img_emb)
114
  #matched_results = util.semantic_search(image_emb, img_emb, 4)[0]
 
115
  image_emb = model.encode([Image.fromarray(search_image.astype('uint8'), 'RGB')], convert_to_tensor=True)
116
  similarity = util.cos_sim(img_emb, image_emb)
117
  return display_matches(similarity)