mlabonne commited on
Commit
a66d9b2
·
verified ·
1 Parent(s): 76f067b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -87,20 +87,24 @@ def make_df(file_path: str, n_rows: int) -> pd.DataFrame:
87
  """
88
  columns = ["Available on the hub", "Model sha", "T", "Type", "Precision",
89
  "Architecture", "Weight type", "Hub ❤️", "Flagged", "MoE"]
90
- ds = pd.read_csv(file_path, encoding='utf-8')
91
  df = (
92
  ds[
93
- (ds["#Params (B)"] == 8.03) &
94
- (ds["Available on the hub"] == True) &
95
- (ds["Flagged"] == False) &
96
- (ds["MoE"] == False) &
97
- (ds["Weight type"] == "Original")
 
 
 
 
98
  ]
99
  .drop(columns=columns)
100
  .drop_duplicates(subset=["Model"])
 
101
  .iloc[:n_rows]
102
  )
103
- print(df)
104
  return df
105
 
106
 
 
87
  """
88
  columns = ["Available on the hub", "Model sha", "T", "Type", "Precision",
89
  "Architecture", "Weight type", "Hub ❤️", "Flagged", "MoE"]
90
+ ds = pd.read_csv("open-llm-leaderboard.csv", encoding='utf-8')
91
  df = (
92
  ds[
93
+ (ds["#Params (B)"] == 8)
94
+ & (ds["Architecture"] == "LlamaForCausalLM")
95
+ & (ds["Available on the hub"] == True)
96
+ & (ds["Flagged"] == True)
97
+ & (~ds["Model"].str.lower().str.contains("yi"))
98
+ & (~ds["Model"].str.lower().str.contains("9b"))
99
+ & (~ds["Model"].str.lower().str.contains("8xqmff94/slm"))
100
+ & (ds["MoE"] == True)
101
+ & (ds["Weight type"] == "Original")
102
  ]
103
  .drop(columns=columns)
104
  .drop_duplicates(subset=["Model"])
105
+ .sort_values(by="MMLU", ascending=False)
106
  .iloc[:n_rows]
107
  )
 
108
  return df
109
 
110