marinap commited on
Commit
800542b
1 Parent(s): 697f3c2

fixed Interface i/o

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,9 +35,9 @@ def find_topk(text):
35
 
36
  vals, inds = sims.topk(top_k)
37
 
38
- top_k_urls = img_df.iloc[inds]['url'].values[0]
39
 
40
- return url2img(top_k_urls)
41
 
42
 
43
 
 
35
 
36
  vals, inds = sims.topk(top_k)
37
 
38
+ top_k_urls = img_df.iloc[inds]['url'].values
39
 
40
+ return [url2img(url) for url in top_k_urls]
41
 
42
 
43