clean up code

io_utils.py

@@ -1,6 +1,5 @@
 import os
 import subprocess
-import gradio as gr
 import yaml
 
 import pipe

text_classification.py

@@ -171,7 +171,6 @@ def infer_output_label_column(
             str(i): id2label_mapping[label]
             for i, label in zip(id2label.keys(), dataset_labels)
         }
-    # print('>>>>> column_mapping >>>>>', column_mapping)
 
     id2label_df = pd.DataFrame(
         {