I do not see you selecting 3 columns, you only selecting data[0] and data[1]. What happened to data[2]? Also how many columns input_df has? I asked you to consider it having a single column.
spam lang_mismatch pii not_appropriate hate_speech sexual_content quality toxicity humor creativity violence