Update space

README.md

@@ -13,7 +13,7 @@ sdk_version: 5.19.0
 ---
 # Model Performance Leaderboard
 
-This is a Hugging Face Space that hosts a leaderboard for comparing model performances across various metrics.
+This is a Hugging Face Space that hosts a leaderboard for comparing model performances across various metrics of TRAIL dataset.
 
 ## Features
 
@@ -62,7 +62,7 @@ You can customize this leaderboard by modifying the `models.json` file:
 
 ```json
 {
-  "title": "Model Performance Leaderboard",
+  "title": "TRAIL Performance Leaderboard",
   "description": "This leaderboard tracks and compares model performance across multiple metrics. Submit your model results to see how they stack up!",
   "metrics": ["accuracy", "f1_score", "precision", "recall"],
   "main_metric": "accuracy"

database.py

@@ -76,10 +76,10 @@ def load_config():
             print("models.json file is empty or missing. Creating with default configuration.")
             # Default configuration
             config = {
-                "title": "Model Leaderboard",
+                "title": "TRAIL Model Leaderboard",
                 "description": "Submit and compare model performances",
-                "metrics": ["accuracy", "f1_score", "precision", "recall"],
-                "main_metric": "accuracy"
+                "metrics": ["Cat. F1", "Loc. Acc", "Joint F1"],
+                "main_metric": "Cat. F1"
             }
             with open("models.json", "w") as f:
                 json.dump(config, f, indent=2)
@@ -88,10 +88,10 @@ def load_config():
         print("Error parsing models.json. Creating with default configuration.")
         # Default configuration if JSON is invalid
         config = {
-            "title": "Model Leaderboard",
+            "title": "TRAIL Model Leaderboard",
             "description": "Submit and compare model performances",
-            "metrics": ["accuracy", "f1_score", "precision", "recall"],
-            "main_metric": "accuracy"
+            "metrics": ["Cat. F1", "Loc. Acc", "Joint F1"],
+            "main_metric": "Cat. F1"
         }
         with open("models.json", "w") as f:
             json.dump(config, f, indent=2)

model

@@ -1,6 +1,6 @@
 {
-  "title": "Model Performance Leaderboard",
+  "title": "TRAIL Performance Leaderboard",
   "description": "This leaderboard tracks and compares model performance across multiple metrics. Submit your model results to see how they stack up!",
-  "metrics": ["accuracy", "f1_score", "precision", "recall"],
-  "main_metric": "accuracy"
+  "metrics": ["Cat. F1", "Loc. Acc", "Joint F1"],
+  "main_metric": "Cat. F1"
 }

models.json

@@ -1,6 +1,6 @@
 {
-    "title": "Model Performance Leaderboard",
+    "title": "TRAIL Performance Leaderboard",
     "description": "This leaderboard tracks and compares model performance across multiple metrics. Submit your model results to see how they stack up!",
-    "metrics": ["accuracy", "f1_score", "precision", "recall"],
-    "main_metric": "accuracy"
+    "metrics": ["Cat. F1", "Loc. Acc", "Joint F1"],
+    "main_metric": "Cat. F1"
   }

setup.py

@@ -21,10 +21,10 @@ def setup():
     if not os.path.exists("models.json") or os.path.getsize("models.json") == 0:
         print("Creating models.json configuration file...")
         config = {
-            "title": "Model Performance Leaderboard",
+            "title": "TRAIL Performance Leaderboard",
             "description": "This leaderboard tracks and compares model performance across multiple metrics. Submit your model results to see how they stack up!",
-            "metrics": ["accuracy", "f1_score", "precision", "recall"],
-            "main_metric": "accuracy"
+            "metrics": ["Cat. F1", "Loc. Acc", "Joint F1"],
+            "main_metric": "Cat. F1"
         }
         with open("models.json", "w") as f:
             json.dump(config, f, indent=2)
@@ -39,8 +39,8 @@ def setup():
             config = {
                 "title": "Model Performance Leaderboard",
                 "description": "This leaderboard tracks and compares model performance across multiple metrics. Submit your model results to see how they stack up!",
-                "metrics": ["accuracy", "f1_score", "precision", "recall"],
-                "main_metric": "accuracy"
+                "metrics": ["Cat. F1", "Loc. Acc", "Joint F1"],
+                "main_metric": "Cat. F1"
             }
             with open("models.json", "w") as f:
                 json.dump(config, f, indent=2)