quick fixes

app.py

@@ -3,6 +3,7 @@ from st_utils import bm25_search, semantic_search, hf_api, paginator
 from huggingface_hub import ModelSearchArguments
 import webbrowser
 from numerize.numerize import numerize
+import math
 
 st.set_page_config(
     page_title="HF Search Engine",
@@ -17,7 +18,7 @@ search_backend = st.sidebar.selectbox(
     ["semantic", "bm25", "hfapi"],
     format_func=lambda x: {"hfapi": "Keyword search", "bm25": "BM25 search", "semantic": "Semantic Search"}[x],
 )
-limit_results = st.sidebar.number_input("Limit results", min_value=0, value=10)
+limit_results = int(st.sidebar.number_input("Limit results", min_value=0, value=10))
 
 st.sidebar.markdown("# Filters")
 args = ModelSearchArguments()
@@ -74,8 +75,8 @@ if search_query != "":
         ):
             col1, col2, col3 = st.columns([5, 1, 1])
             col1.metric("Model", hit["modelId"])
-            col2.metric("N° downloads", numerize(hit["downloads"]))
-            col3.metric("N° likes", numerize(hit["likes"]))
+            col2.metric("N° downloads", numerize(hit["downloads"]) if not math.isnan(hit["downloads"]) else "N/A")
+            col3.metric("N° likes", numerize(hit["likes"]) if not math.isnan(hit["likes"]) else "N/A")
             st.button(
                 f"View model on 🤗",
                 on_click=lambda hit=hit: webbrowser.open(f"https://huggingface.co/{hit['modelId']}"),

requirements.txt

@@ -1,5 +1,5 @@
-git+https://github.com/NouamaneTazi/hf_search
 pandas
 streamlit
 huggingface_hub
 numerize
+git+https://github.com/NouamaneTazi/hf_search

st_utils.py

@@ -68,8 +68,8 @@ def bm25_search(query, limit=5, filters={}):
         {
             "modelId": hit["modelId"],
             "tags": hit["tags"],
-            "downloads": int(hit["downloads"]),
-            "likes": int(hit["likes"]),
+            "downloads": hit["downloads"],
+            "likes": hit["likes"],
             "readme": hit.get("readme", None),
         }
         for hit in hits