Update app.py

app.py

@@ -21,11 +21,17 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 block = gr.Blocks()
 api = HfApi(token=HF_TOKEN)
 
-# custom_css = """
+
 # .tab-buttons button {
 #     font-size: 20px;
 # }
-# """
+
+custom_css = """
+/* Full width space */
+.gradio-container {
+  max-width: 95%!important;
+}
+"""
 
 hivex_envs = [
     {
@@ -64,7 +70,15 @@ def restart():
 def download_leaderboard_dataset():
     path = snapshot_download(repo_id=DATASET_REPO_ID, repo_type="dataset")
     return path
+    
 
+def get_total_models():
+    total_models = 0
+    for hivex_env in hivex_envs:
+        model_ids = get_model_ids(hivex_env["hivex_env"])
+        total_models += len(model_ids)
+    return total_models
+    
 
 def get_model_ids(hivex_env):
     api = HfApi()
@@ -160,12 +174,11 @@ def get_data(rl_env, task, path) -> pd.DataFrame:
 
 run_update_dataset()
 
-# block = gr.Blocks(css=custom_css)
-block = gr.Blocks()
+block = gr.Blocks(css=custom_css)
 with block:
     with gr.Row(elem_id="header-row"):
-        # TITLE + "<p>Total models: " + str(len(HARD_LEADERBOARD_DF))+ "</p>"
-        gr.HTML("<h1>HIVEX-Leaderboard</h1>")
+        total_models = get_total_models()
+        gr.HTML(f"<h1>HIVEX-Leaderboard</h1><p>Total models: {total_models}</p>")
 
     path_ = download_leaderboard_dataset()
     # gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")