fix model search

app.py

@@ -224,7 +224,6 @@ def change_tab(query_param: str):
 # Searching and filtering
 def update_table(
     hidden_df: pd.DataFrame,
-    current_columns_df: pd.DataFrame,
     columns: list,
     type_query: list,
     precision_query: str,
@@ -233,16 +232,7 @@ def update_table(
     query: str,
 ):
     filtered_df = filter_models(hidden_df, type_query, size_query, precision_query, show_deleted)
-    final_df = []
-    if query != "":
-        queries = query.split(";")
-        for _q in queries:
-            if _q != "":
-                temp_filtered_df = search_table(filtered_df, _q)
-                if len(temp_filtered_df) > 0:
-                    final_df.append(temp_filtered_df)
-        if len(final_df) > 0:
-            filtered_df = pd.concat(final_df).drop_duplicates()
+    filtered_df = filter_queries(query, filtered_df)
     df = select_columns(filtered_df, columns)
     return df
 
@@ -250,7 +240,6 @@ def update_table(
 def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
     return df[(df[AutoEvalColumn.dummy.name].str.contains(query, case=False))]
 
-
 def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
     always_here_cols = [
         AutoEvalColumn.model_type_symbol.name,
@@ -274,6 +263,23 @@ NUMERIC_INTERVALS = {
 }
 
 
+def filter_queries(query: str, filtered_df: pd.DataFrame):
+    """Added by Abishek"""
+    final_df = []
+    if query != "":
+        queries = [q.strip() for q in query.split(";")]
+        for _q in queries:
+            _q = _q.strip()
+            if _q != "":
+                temp_filtered_df = search_table(filtered_df, _q)
+                if len(temp_filtered_df) > 0:
+                    final_df.append(temp_filtered_df)
+        if len(final_df) > 0:
+            filtered_df = pd.concat(final_df)
+            filtered_df = filtered_df.drop_duplicates(subset=[AutoEvalColumn.model.name, AutoEvalColumn.precision.name, AutoEvalColumn.revision.name])
+
+    return filtered_df
+
 def filter_models(
     df: pd.DataFrame, type_query: list, size_query: list, precision_query: list, show_deleted: bool
 ) -> pd.DataFrame:
@@ -409,7 +415,6 @@ with demo:
                 update_table,
                 [
                     hidden_leaderboard_table_for_search,
-                    leaderboard_table,
                     shown_columns,
                     filter_columns_type,
                     filter_columns_precision,
@@ -423,7 +428,6 @@ with demo:
                 update_table,
                 [
                     hidden_leaderboard_table_for_search,
-                    leaderboard_table,
                     shown_columns,
                     filter_columns_type,
                     filter_columns_precision,
@@ -438,7 +442,6 @@ with demo:
                 update_table,
                 [
                     hidden_leaderboard_table_for_search,
-                    leaderboard_table,
                     shown_columns,
                     filter_columns_type,
                     filter_columns_precision,
@@ -453,7 +456,6 @@ with demo:
                 update_table,
                 [
                     hidden_leaderboard_table_for_search,
-                    leaderboard_table,
                     shown_columns,
                     filter_columns_type,
                     filter_columns_precision,
@@ -468,7 +470,6 @@ with demo:
                 update_table,
                 [
                     hidden_leaderboard_table_for_search,
-                    leaderboard_table,
                     shown_columns,
                     filter_columns_type,
                     filter_columns_precision,
@@ -483,7 +484,6 @@ with demo:
                 update_table,
                 [
                     hidden_leaderboard_table_for_search,
-                    leaderboard_table,
                     shown_columns,
                     filter_columns_type,
                     filter_columns_precision,

model_info_cache.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8fcaa2a3e1ac6a5559471547af5de4e3ccd49673ad5525890726e65cd90cfe62
-size 3620752
+oid sha256:94e311e2414e80b8eb5e50844c2e79daa4bd3bb6be516fc2448bd05242d125f9
+size 3656702

model_size_cache.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:75d1f64589459eb64e3a50987bf05ed3656248102d1fe2f6c98a008020945840
-size 74321
+oid sha256:4aedc91b51cf257cbe3e26a1fdd99e19250bacfa619a64dd85e67d4ff383130f
+size 75455

src/display_models/get_model_metadata.py

@@ -40,7 +40,7 @@ def get_model_infos_from_hub(leaderboard_data: List[dict]):
             try:
                 model_info = api.model_info(model_name)
                 model_info_cache[model_name] = model_info
-            except huggingface_hub.utils._errors.RepositoryNotFoundError:
+            except (huggingface_hub.utils._errors.RepositoryNotFoundError, huggingface_hub.utils._errors.HfHubHTTPError):
                 print("Repo not found!", model_name)
                 model_data[AutoEvalColumn.license.name] = None
                 model_data[AutoEvalColumn.likes.name] = None