WebashalarForML commited on
Commit
3426b11
·
verified ·
1 Parent(s): 6c80eb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -56,9 +56,13 @@ def match_image():
56
  results = []
57
  for b64_img in data["images"]:
58
  try:
59
- img = decode_base64_image(b64_img)
60
- query_embed = np.array(clip_embd.embed_image([img]))
61
 
 
 
 
 
62
  sims = cosine_similarity(query_embed, image_embeds)[0]
63
  best_idx = np.argmax(sims)
64
 
 
56
  results = []
57
  for b64_img in data["images"]:
58
  try:
59
+ # Convert Base64 → BytesIO
60
+ b_io = BytesIO(base64.b64decode(b64_img))
61
 
62
+ # Embed the query image
63
+ query_embed = np.array(clip_embd.embed_image([b_io]))
64
+
65
+ # Cosine similarity with stored embeddings
66
  sims = cosine_similarity(query_embed, image_embeds)[0]
67
  best_idx = np.argmax(sims)
68