marinap commited on
Commit
904f0e1
1 Parent(s): 5c329ad

troubleshooting image csv

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def find_topk(text):
35
  sims = F.cosine_similarity(text_embedding, embeddings)
36
 
37
  vals, inds = sims.topk(top_k)
38
-
39
  top_k_urls = img_df.iloc[inds]['photo_image_url'].values
40
 
41
  print('top_k_urls', top_k_urls)
 
35
  sims = F.cosine_similarity(text_embedding, embeddings)
36
 
37
  vals, inds = sims.topk(top_k)
38
+ print(img_df)
39
  top_k_urls = img_df.iloc[inds]['photo_image_url'].values
40
 
41
  print('top_k_urls', top_k_urls)