Update app.py

app.py

@@ -28,6 +28,7 @@ def add_new_eval(
     model_link: str,
     team_name: str,
     contact_email: str,
+    access_type: str,
     model_publish: str,
     model_resolution: str,
     model_fps: str,
@@ -116,6 +117,8 @@ def add_new_eval(
 
     new_data.append(contact_email.replace(',',' and ')) # Add contact email [private]
     new_data.append(update_time)  # Add the update time
+    new_data.append(team_name)
+    new_data.append(access_type)
 
     csv_data.loc[col] = new_data
     csv_data = csv_data.to_csv(CSV_DIR, index=False)
@@ -132,6 +135,7 @@ def add_new_eval_i2v(
     model_link: str,
     team_name: str,
     contact_email: str,
+    access_type: str,
     model_publish: str,
     model_resolution: str,
     model_fps: str,
@@ -225,7 +229,7 @@ def add_new_eval_i2v(
     "Video-Image Background Consistency",
     "Subject Consistency",
     "Background Consistency",
-    # "Temporal Flickering",
+    "Temporal Flickering",
     "Motion Smoothness",
     "Dynamic Degree",
     "Aesthetic Quality",
@@ -243,6 +247,8 @@ def add_new_eval_i2v(
 
     new_data.append(contact_email.replace(',',' and ')) # Add contact email [private]
     new_data.append(update_time)  # Add the update time
+    new_data.append(team_name)
+    new_data.append(access_type)
 
     csv_data.loc[col] = new_data
     csv_data = csv_data.to_csv(I2V_DIR , index=False)
@@ -339,7 +345,7 @@ def get_final_score(df, selected_columns):
 def get_final_score_i2v(df, selected_columns):
     normalize_df = get_normalized_i2v_df(df)
     try:
-        for name in normalize_df.drop('Model Name (clickable)', axis=1).drop('Video-Text Camera Motion', axis=1).drop("Sampled by", axis=1).drop('Mail', axis=1).drop('Date',axis=1).drop("Evaluated by", axis=1).drop("Accessibility", axis=1):
+        for name in normalize_df.drop('Model Name (clickable)', axis=1).drop("Sampled by", axis=1).drop('Mail', axis=1).drop('Date',axis=1).drop("Evaluated by", axis=1).drop("Accessibility", axis=1):
             normalize_df[name] = normalize_df[name]*DIM_WEIGHT_I2V[name]
     except:
         for name in normalize_df.drop('Model Name (clickable)', axis=1).drop("Sampled by", axis=1).drop('Mail', axis=1).drop('Date',axis=1):
@@ -887,6 +893,8 @@ with block:
                     revision_name_textbox_i2v = gr.Textbox(
                         label="Revision Model Name(Optional)", placeholder="If you need to update the previous results, please fill in this line"
                     )
+                    access_type_i2v = gr.Dropdown(["Open Source", "Ready to Open Source", "API", "Close"], label="Please select the way user can access your model. You can update the content by revision_name, or contact the VBench Team.")
+
 
                 with gr.Column():
                     model_link_i2v = gr.Textbox(
@@ -927,7 +935,7 @@ with block:
                         team_name_i2v,
                         contact_email_i2v,
                         release_time_i2v,
-                        access_type,
+                        access_type_i2v,
                         model_resolution_i2v,
                         model_fps_i2v,
                         model_frame_i2v,