fix

app.py

@@ -5,6 +5,7 @@ import matplotlib.pyplot as plt
 import bar_chart_race as bcr
 import pandas as pd
 import gradio as gr
+import requests
 import os
 
 
@@ -15,10 +16,15 @@ def restart_space():
     )
 
 
-# read in the data
-open_llm_race_dataset = pd.read_csv(
-    "https://huggingface.co/datasets/IlyasMoutawwakil/open-llm-race-dataset/resolve/main/open-llm-race-dataset.csv"
-)
+if not os.path.exists("open-llm-race-dataset.csv"):
+    file = requests.get(
+        "https://huggingface.co/datasets/IlyasMoutawwakil/open-llm-race-dataset/resolve/main/open-llm-race-dataset.csv"
+    ).content
+    with open("open-llm-race-dataset.csv") as f:
+        f.write(file)
+
+open_llm_race_dataset = pd.read_csv("open-llm-race-dataset.csv")
+
 # resample for ever model to a daily frequency
 open_llm_race_dataset["date"] = pd.to_datetime(open_llm_race_dataset["date"])
 open_llm_race_dataset = (
@@ -44,7 +50,7 @@ open_llm_race_dataset["type"] = open_llm_race_dataset["model"].apply(
 )
 
 
-def get_bar_chart(model_type: str):
+def get_bar_chart(model_type: str, top_n: int = 10):
     fig, ax = plt.subplots(figsize=(12, 6))
     ax.set_xlim(0, 100)
     subset = open_llm_race_dataset[open_llm_race_dataset["type"] == model_type]
@@ -68,16 +74,15 @@ def get_bar_chart(model_type: str):
 demo = gr.Blocks()
 with demo:
     # leaderboard title
-    gr.HTML("<h1>Open-LLM Race 🏃‍♂️</h1>")
+    gr.HTML("<center><h1>LLM Bar Race 📊🏃‍♂️</h1></center>")
 
     with gr.Tabs():
         with gr.TabItem(label="Pretrained Models"):
             get_bar_chart(ModelType.PT.name)
         with gr.TabItem(label="Instructions Finetuend Models"):
             get_bar_chart(ModelType.IFT.name)
-
         with gr.TabItem(label="RLHF Models"):
-            get_bar_chart(ModelType.RL.name)
+            get_bar_chart(ModelType.RL.name, top_n=4)
 
 
 # Restart space every hour