Update space

app.py

@@ -221,8 +221,8 @@ def save_submission(submission_data, csv_file):
         csv_file: The uploaded CSV file object
     """
     # Create folder name from model name and team name
-    model_name_clean = sanitize_name(submission_data['method_name'])
-    team_name_clean = sanitize_name(submission_data['team_name'])
+    model_name_clean = sanitize_name(submission_data['Method Name'])
+    team_name_clean = sanitize_name(submission_data['Team Name'])
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     
     # Create folder name: model_name_team_name
@@ -259,7 +259,7 @@ def save_submission(submission_data, csv_file):
         json.dump({
             "latest_submission": timestamp,
             "status": "pending_review",
-            "method_name": submission_data['method_name']
+            "method_name": submission_data['Method Name']
         }, f, indent=4)
     
     return submission_id
@@ -282,15 +282,15 @@ def update_leaderboard_data(submission_data):
     
     # Prepare new row data
     new_row = {
-        'Method': submission_data['method_name'],  # Only use method name in table
-        f'STARK-{submission_data["dataset"].upper()}_Hit@1': submission_data['results']['hit@1'],
-        f'STARK-{submission_data["dataset"].upper()}_Hit@5': submission_data['results']['hit@5'],
-        f'STARK-{submission_data["dataset"].upper()}_R@20': submission_data['results']['recall@20'],
-        f'STARK-{submission_data["dataset"].upper()}_MRR': submission_data['results']['mrr']
+        'Method': submission_data['Method Name'],  # Only use method name in table
+        f'STARK-{submission_data["Dataset"].upper()}_Hit@1': submission_data['results']['hit@1'],
+        f'STARK-{submission_data["Dataset"].upper()}_Hit@5': submission_data['results']['hit@5'],
+        f'STARK-{submission_data["Dataset"].upper()}_R@20': submission_data['results']['recall@20'],
+        f'STARK-{submission_data["Dataset"].upper()}_MRR': submission_data['results']['mrr']
     }
     
     # Check if method already exists
-    method_mask = df_to_update['Method'] == submission_data['method_name']
+    method_mask = df_to_update['Method'] == submission_data['Method Name']
     if method_mask.any():
         # Update existing row
         for col in new_row: