siro1 commited on
Commit
893c5ca
·
1 Parent(s): 480870f
Files changed (1) hide show
  1. app.py +16 -6
app.py CHANGED
@@ -14,7 +14,11 @@ dataframe: pd.DataFrame = None
14
  def load_dataframe(dataset=None) -> pd.DataFrame:
15
  global dataframe
16
  if dataset is None:
17
- dataset = load_dataset(DATASET_ID, split="train")
 
 
 
 
18
  results = dataset.map(
19
  lambda item: {
20
  "Team": item["team"],
@@ -127,8 +131,15 @@ def refresh_leaderboard(push_to_hub: bool = True):
127
  dataset.push_to_hub(DATASET_ID, token=HF_TOKEN)
128
 
129
 
 
 
 
 
 
 
130
  # Create Gradio interface
131
  def create_interface():
 
132
  with gr.Blocks(title="AMD vLLM Benchmark Leaderboard") as demo:
133
  gr.Markdown("# AMD vLLM Benchmark Leaderboard")
134
  gr.Markdown(
@@ -179,14 +190,14 @@ def create_interface():
179
  )
180
 
181
  # Auto-refresh every 30 seconds
182
- timer = gr.Timer(30) # 30 seconds interval
183
  timer.tick(
184
  fn=refresh_leaderboard,
185
  )
186
 
187
- timer = gr.Timer(30)
188
- timer.tick(
189
- fn=lambda: dataframe,
190
  outputs=leaderboard_table,
191
  )
192
 
@@ -202,5 +213,4 @@ if __name__ == "__main__":
202
  server_name="0.0.0.0",
203
  server_port=7860,
204
  share=True,
205
- max_threads=40,
206
  )
 
14
  def load_dataframe(dataset=None) -> pd.DataFrame:
15
  global dataframe
16
  if dataset is None:
17
+ # Force download of latest data by disabling cache
18
+ dataset = load_dataset(
19
+ DATASET_ID, split="train", download_mode="force_redownload"
20
+ )
21
+ print(f"Loaded dataset: {len(dataset)}")
22
  results = dataset.map(
23
  lambda item: {
24
  "Team": item["team"],
 
131
  dataset.push_to_hub(DATASET_ID, token=HF_TOKEN)
132
 
133
 
134
+ def get_leaderboard():
135
+ global dataframe
136
+ print(f"Getting leaderboard: {len(dataframe)}")
137
+ return dataframe
138
+
139
+
140
  # Create Gradio interface
141
  def create_interface():
142
+ global dataframe
143
  with gr.Blocks(title="AMD vLLM Benchmark Leaderboard") as demo:
144
  gr.Markdown("# AMD vLLM Benchmark Leaderboard")
145
  gr.Markdown(
 
190
  )
191
 
192
  # Auto-refresh every 30 seconds
193
+ timer = gr.Timer(15) # 30 seconds interval
194
  timer.tick(
195
  fn=refresh_leaderboard,
196
  )
197
 
198
+ data_timer = gr.Timer(15)
199
+ data_timer.tick(
200
+ fn=get_leaderboard,
201
  outputs=leaderboard_table,
202
  )
203
 
 
213
  server_name="0.0.0.0",
214
  server_port=7860,
215
  share=True,
 
216
  )