Update app.py

app.py

@@ -56,20 +56,26 @@ def search(query):
     query = query.lower()
     query = ' '.join([query[i:i+4] for i in range(len(query)-3)])
     scores = model.predict({"Name": query})
-    K = min(2*max_posts, len(scores) - 1)
-    sorted_post_ids = scores.argsort()[-K:][::-1]
-    print(sorted_post_ids)
-    sorted_ids = []
-    relevant_posts = []
-    count = 0
-    for pid in sorted_post_ids:
-        if pid in recipe_id_to_row_num:
-            relevant_posts.append(df.iloc[recipe_id_to_row_num[pid]])
-            sorted_ids.append(pid)
-        count += 1
-        if count==max_posts:
-            break
-    ##
+    ####
+    sorted_ids = scores.argsort()[-max_posts:][::-1]
+    relevant_posts = [
+        df.iloc[pid] for pid in sorted_ids
+    ]
+    ####
+    # K = min(2*max_posts, len(scores) - 1)
+    # sorted_post_ids = scores.argsort()[-K:][::-1]
+    # print(sorted_post_ids)
+    # sorted_ids = []
+    # relevant_posts = []
+    # count = 0
+    # for pid in sorted_post_ids:
+    #     if pid in recipe_id_to_row_num:
+    #         relevant_posts.append(df.iloc[recipe_id_to_row_num[pid]])
+    #         sorted_ids.append(pid)
+    #     count += 1
+    #     if count==max_posts:
+    #         break
+    ####
     header = [gr.Markdown.update(visible=True)]
     boxes = [
         gr.Box.update(visible=True)