marcusinthesky commited on
Commit
e05a8b4
·
1 Parent(s): 4b3d8d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -30,7 +30,10 @@ def search(image=None, query=None):
30
  image_embed = np.array(pipe.encode(image))
31
 
32
  if query is not None and image is not None:
33
- embed = text_embed*(image_embed**0.2)
 
 
 
34
  elif image is not None:
35
  embed = image_embed
36
  else:
 
30
  image_embed = np.array(pipe.encode(image))
31
 
32
  if query is not None and image is not None:
33
+ text_embed_normalized = text_embed / np.linalg.norm(text_embed)
34
+ image_embed_normalized = image_embed / np.linalg.norm(image_embed)
35
+ embed = text_embed_normalized + image_embed_normalized
36
+ embed = embed / np.linalg.norm(embed) # we renormalize
37
  elif image is not None:
38
  embed = image_embed
39
  else: