Update constants.py

constants.py

@@ -49,6 +49,35 @@ TASK_INFO = [
     "overall consistency"
 ]
 
+
+MODEL_INFO_2 = [
+    "Model Name (clickable)",
+    "Sampled by",
+    "Evaluated by",
+    "Accessibility",
+    "Date"
+]
+TASK_INFO_2 = [
+    "Human Anatomy",
+    "Human Clothes",
+    "Human Identity",
+    "Composition",
+    "Diversity",
+    "Mechanics",
+    "Material",
+    "Themotics",
+    "Multi-view Consistency",
+    "Dynamic Spatial Relationship",
+    "Dynamic Attribute",    
+    "Motion Order Understanding",
+    "Human Interaction",
+    "Complex Landscape",
+    "Complex Plot", 
+    "Camera Motion",
+    "Motion Rationality",
+    "Instance Preservation"
+    ]
+
 DEFAULT_INFO = [
     "subject consistency",
     "background consistency",
@@ -165,6 +194,7 @@ I2V_QUALITY_WEIGHT = 1.0
 
 DATA_TITILE_TYPE = ['markdown', 'markdown', 'markdown', 'markdown', 'markdown', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number']
 I2V_TITILE_TYPE =  ['markdown', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number']
+VBENCH2_TITLE_TYPE = ['markdown', 'markdown', 'markdown', 'markdown', 'markdown', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number', 'number']
 
 SUBMISSION_NAME = "vbench_leaderboard_submission"
 SUBMISSION_URL = os.path.join("https://huggingface.co/datasets/Vchitect/", SUBMISSION_NAME)
@@ -173,8 +203,10 @@ QUALITY_DIR = "./vbench_leaderboard_submission/quality.csv"
 I2V_DIR = "./vbench_leaderboard_submission/i2v_results.csv"
 LONG_DIR = "./vbench_leaderboard_submission/long_debug.csv"
 INFO_DIR = "./vbench_leaderboard_submission/model_info.csv"
+VBENCH2_DIR = "./vbench_leaderboard_submission/results2_0.csv"
 
 COLUMN_NAMES = MODEL_INFO + TASK_INFO
+COLUMN_NAMES_2 = MODEL_INFO_2 + TASK_INFO_2
 COLUMN_NAMES_QUALITY = MODEL_INFO_TAB_QUALITY + QUALITY_TAB
 COLUMN_NAMES_I2V = MODEL_INFO_TAB_I2V + I2V_TAB
 
@@ -263,4 +295,4 @@ NORMALIZE_DIC_I2V = {
     "Aesthetic Quality":{"Min": 0.0, "Max": 1.0},
     "Imaging Quality":{"Min": 0.0, "Max": 1.0},
     "Temporal Flickering":{"Min":0.6293, "Max": 1.0}
-}
+}