updated scripts

app.py

@@ -12,9 +12,32 @@ import src.submission.submit as submit
 import os
 TOKEN = os.environ.get("H4_TOKEN", None)
 print("TOKEN", TOKEN)
+
+def ui_snapshot_download(repo_id, local_dir, repo_type, tqdm_class, etag_timeout):
+    try:
+        print(local_dir)
+        snapshot_download(repo_id=repo_id, local_dir=local_dir, repo_type=repo_type, tqdm_class=tqdm_class, etag_timeout=etag_timeout)
+    except Exception as e:
+        restart_space()
+
 def restart_space():
     envs.API.restart_space(repo_id=envs.REPO_ID, token=TOKEN)
 
+def init_space():
+    dataset_df = get_dataset_summary_table(file_path='blog/Hallucination-Leaderboard-Summary.csv')
+
+    if socket.gethostname() not in {'neuromancer'}:
+        # sync model_type with open-llm-leaderboard
+        ui_snapshot_download(repo_id=QUEUE_REPO, local_dir=EVAL_REQUESTS_PATH, repo_type="dataset", tqdm_class=None, etag_timeout=30)
+        ui_snapshot_download(repo_id=RESULTS_REPO, local_dir=EVAL_RESULTS_PATH, repo_type="dataset", tqdm_class=None, etag_timeout=30)
+    raw_data, original_df = get_leaderboard_df(EVAL_RESULTS_PATH, EVAL_REQUESTS_PATH, "", COLS, BENCHMARK_COLS)
+
+    finished_eval_queue_df, running_eval_queue_df, pending_eval_queue_df = get_evaluation_queue_df(EVAL_REQUESTS_PATH, EVAL_COLS)
+    return dataset_df, original_df, finished_eval_queue_df, running_eval_queue_df, pending_eval_queue_df
+
+dataset_df, original_df, finished_eval_queue_df, running_eval_queue_df, pending_eval_queue_df = init_space()
+
+
 # try:
 #     print(envs.EVAL_REQUESTS_PATH)
 #     snapshot_download(
@@ -30,7 +53,9 @@ def restart_space():
 # except Exception:
 #     restart_space()
 
-raw_data, original_df = populate.get_leaderboard_df(envs.RESULTS_REPO, envs.QUEUE_REPO, utils.COLS, utils.BENCHMARK_COLS)
+# raw_data, original_df = populate.get_leaderboard_df(envs.RESULTS_REPO, envs.QUEUE_REPO, utils.COLS, utils.BENCHMARK_COLS)
+
+
 leaderboard_df = original_df.copy()
 
 (