hysts HF staff commited on
Commit
28bab67
1 Parent(s): 245d717
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -88,13 +88,13 @@ MAX_MODEL_SIZE = ORIGINAL_DF["#Params (B)"].max()
88
 
89
  def filter_models(
90
  df: pd.DataFrame,
91
- type_query: list,
92
- size_query: list,
93
- precision_query: list,
94
- add_special_tokens_query: list,
95
- num_few_shots_query: list,
96
- version_query: list,
97
- vllm_query: list,
98
  ) -> pd.DataFrame:
99
  print(f"Initial df shape: {df.shape}")
100
  print(f"Initial df content:\n{df}")
 
88
 
89
  def filter_models(
90
  df: pd.DataFrame,
91
+ type_query: list[str],
92
+ size_query: list[str],
93
+ precision_query: list[str],
94
+ add_special_tokens_query: list[str],
95
+ num_few_shots_query: list[str],
96
+ version_query: list[str],
97
+ vllm_query: list[str],
98
  ) -> pd.DataFrame:
99
  print(f"Initial df shape: {df.shape}")
100
  print(f"Initial df content:\n{df}")