fix

app.py

@@ -1,21 +1,26 @@
-# from apscheduler.schedulers.background import BackgroundScheduler
+from apscheduler.schedulers.background import BackgroundScheduler
 from model_types import MODEL_TYPES, ModelType
-
-# from huggingface_hub import HfApi
+from huggingface_hub import HfApi
 import matplotlib.pyplot as plt
 import bar_chart_race as bcr
 import pandas as pd
 import gradio as gr
+import os
 
-# import os
 
+def restart_space():
+    HfApi(token=os.environ.get("HF_TOKEN", None)).restart_space(
+        repo_id="IlyasMoutawwakil/llm-bar-race",
+        token=os.environ.get("HF_TOKEN", None),
+    )
 
-# def restart_space():
-#     HfApi(token=os.environ.get("HF_TOKEN", None)).restart_space(
-#         repo_id="IlyasMoutawwakil/llm-bar-race",
-#         token=os.environ.get("HF_TOKEN", None),
-#     )
 
+scheduler = BackgroundScheduler()
+scheduler.add_job(
+    func=restart_space,
+    trigger="interval",
+    seconds=60 * 60 * 6,
+)
 
 open_llm_race_dataset = pd.read_parquet(
     "https://huggingface.co/datasets/IlyasMoutawwakil/llm-race-dataset/resolve/main/llm-race-dataset.parquet",
@@ -32,9 +37,9 @@ open_llm_race_dataset = (
 )
 # filter
 open_llm_race_dataset["date"] = open_llm_race_dataset["date"].dt.strftime("%Y-%m-%d")
-open_llm_race_dataset = open_llm_race_dataset[
-    open_llm_race_dataset["date"] >= "2023-07-10"
-]
+# open_llm_race_dataset = open_llm_race_dataset[
+#     open_llm_race_dataset["date"] >= "2023-07-10"
+# ]
 open_llm_race_dataset = open_llm_race_dataset[["date", "score", "model"]]
 # drop nan values
 open_llm_race_dataset.dropna(inplace=True)
@@ -53,7 +58,7 @@ def get_bar_chart(model_type: str, top_n: int = 10, title: str = ""):
 
     subset = open_llm_race_dataset[open_llm_race_dataset["type"] == model_type]
     subset = subset.pivot(index="date", columns="model", values="score")
-    subset.fillna(0, inplace=True)
+    # subset.fillna(0, inplace=True)
     fig = bcr.bar_chart_race(
         subset,
         fig=fig,
@@ -84,8 +89,9 @@ with demo:
         with gr.TabItem(label="Instructions Finetuned Models"):
             get_bar_chart(ModelType.IFT.name, title="Instructions Finetuned Models")
         with gr.TabItem(label="RLHF Models"):
-            get_bar_chart(ModelType.RL.name, top_n=4, title="RLHF Models")
+            get_bar_chart(ModelType.RL.name, title="RLHF Models")
         with gr.TabItem(label="Finetuned Models"):
             get_bar_chart(ModelType.FT.name, title="Finetuned Models")
 
+scheduler.start()
 demo.queue(concurrency_count=10).launch()