Update app.py

app.py

@@ -37,7 +37,7 @@ hf_fs = HfFileSystem(token=HF_TOKEN)
 input_repo_path = 'datasets/emvecchi/annotate-pilot'
 output_repo_path = 'datasets/emvecchi/annotate-pilot'
 to_annotate_file_name = 'to_annotate.csv'  # CSV file to annotate
-COLS_TO_SAVE = ['comment_id']
+COLS_TO_SAVE = ['comment_id','comment','confidence_score']
 
 agreement_labels = ['strongly disagree', 'disagree', 'neither agree no disagree', 'agree', 'strongly agree']
 quality_labels = ['very poor', 'poor', 'acceptable', 'good', 'very good']