Update classify.py

classify.py

@@ -10,7 +10,7 @@ from utils import setup_logging, load_system_prompt, load_queries, load_checkpoi
 
 def run_inference(cfg, df_input, ckpt_df, all_rows, total, future_to_idx, start_time, logger):
     completed = 0
-    print("before: \n", df_input.columns)
+    # print("before: \n", df_input.columns)
     with tqdm.tqdm(total=total, desc="Classifying") as pbar:
         for future in as_completed(future_to_idx):
             try:
@@ -43,7 +43,7 @@ def run_inference(cfg, df_input, ckpt_df, all_rows, total, future_to_idx, start_
                 else:
                     combined = batch_df
                 save_checkpoint(combined, cfg)
-    print("after: \n", all_rows)
+    # print("after: \n", all_rows)
 
     # --- Final output ---
     final_df = pd.concat([df_input, pd.DataFrame(all_rows).drop(columns=[cfg.data[cfg.level].query_column])], axis=1)