quick fixes

pages/search_engine.py

@@ -131,7 +131,7 @@ def page():
                 col2.metric("N° downloads", numerize(hit["downloads"]))
                 col3.metric("N° likes", numerize(hit["likes"]))
                 st.button(f"View model on 🤗", on_click=lambda hit=hit: webbrowser.open(f"https://huggingface.co/{hit['modelId']}"), key=hit["modelId"])
-                st.markdown(f"**Tags:** {'  •  '.join(hit['tags'])}")
+                st.write(f"**Tags:** {'  •  '.join(hit['tags'])}")
 
                 if hit["readme"]:
                     with st.expander("See README"):

server/api.py

@@ -56,8 +56,7 @@ def semantic_search():
     print("filters", filters)
     print("limit", limit)
 
-    # TODO: filters
-    hits = hf_search(query=query, method="retrieve & rerank", limit=limit)
+    hits = hf_search(query=query, method="retrieve & rerank", limit=limit, filters=filters)
     hits = [
         {
             "modelId": hit["modelId"],
@@ -92,8 +91,7 @@ def bm25_search():
         }
         for hit in hits
     ]
-    pprint(hits)
-
+    hits = [hits[i] for i in range(len(hits)) if hits[i]["modelId"] not in [h["modelId"] for h in hits[:i]]] # unique hits
     return json.dumps({"value": hits, "count": len(hits)})