leedoming commited on
Commit
7917a23
1 Parent(s): ed40fd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def get_average_embedding(main_image_url, additional_image_urls):
83
 
84
  if embeddings:
85
  avg_embedding = np.mean(embeddings, axis=0)
86
- return avg_embedding.tolist() if isinstance(avg_embedding, np.ndarray) else avg_embedding
87
  else:
88
  return None
89
 
 
83
 
84
  if embeddings:
85
  avg_embedding = np.mean(embeddings, axis=0)
86
+ return avg_embedding if isinstance(avg_embedding, np.ndarray) else avg_embedding
87
  else:
88
  return None
89