tharunkr24 commited on
Commit
b2a2ad8
·
1 Parent(s): 3fb7896

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -56,12 +56,13 @@ def search(query):
56
  scores = model.predict({"Name": query.lower()})
57
  K = min(2*max_posts, len(scores) - 1)
58
  sorted_post_ids = scores.argsort()[-K:][::-1]
59
- count = 0
60
  relevant_posts = []
 
61
  for pid in sorted_post_ids:
62
  if pid in recipe_id_to_row_num:
63
  relevant_posts.append(df.iloc[recipe_id_to_row_num[pid]])
64
- ##
65
  count += 1
66
  if count==max_posts:
67
  break
@@ -110,7 +111,7 @@ def search(query):
110
  toggles +
111
  matches +
112
  bodies +
113
- [sorted_post_ids]
114
  )
115
 
116
 
 
56
  scores = model.predict({"Name": query.lower()})
57
  K = min(2*max_posts, len(scores) - 1)
58
  sorted_post_ids = scores.argsort()[-K:][::-1]
59
+ sorted_ids = []
60
  relevant_posts = []
61
+ count = 0
62
  for pid in sorted_post_ids:
63
  if pid in recipe_id_to_row_num:
64
  relevant_posts.append(df.iloc[recipe_id_to_row_num[pid]])
65
+ sorted_ids.append(pid)
66
  count += 1
67
  if count==max_posts:
68
  break
 
111
  toggles +
112
  matches +
113
  bodies +
114
+ [sorted_ids]
115
  )
116
 
117