[fix] COLUMN_NAMES_I2V

app.py

@@ -312,7 +312,6 @@ def on_filter_model_size_method_change_quality(selected_columns):
     return filter_component#.value
 
 def on_filter_model_size_method_change_i2v(selected_columns):
-
     updated_data = get_all_df_i2v(selected_columns, I2V_DIR)
     selected_columns = [item for item in I2V_TAB if item in selected_columns]
     present_columns = MODEL_INFO_TAB_I2V + selected_columns
@@ -320,9 +319,7 @@ def on_filter_model_size_method_change_i2v(selected_columns):
     updated_data = updated_data.sort_values(by="Selected Score", ascending=False)
     updated_data = convert_scores_to_percentage(updated_data)
     updated_headers = present_columns
-    update_datatype = [DATA_TITILE_TYPE[COLUMN_NAMES.index(x)] for x in updated_headers]
-    import ipdb
-    ipdb.set_trace()
+    update_datatype = [DATA_TITILE_TYPE[COLUMN_NAMES_I2V.index(x)] for x in updated_headers]
     # print(updated_data,present_columns,update_datatype)
     filter_component = gr.components.Dataframe(
         value=updated_data,