Spaces:
Sleeping
Sleeping
Final
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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(
|
| 183 |
timer.tick(
|
| 184 |
fn=refresh_leaderboard,
|
| 185 |
)
|
| 186 |
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
fn=
|
| 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 |
)
|