Select columns for summarization preview

yourbench_space/utils.py

@@ -98,7 +98,7 @@ def update_dataset(stages: list, hf_org: str, hf_prefix: str, oauth_token: gr.OA
     if STAGE_DISPLAY_MAP["summarization"] in stages:
         summarization_ds = load_dataset(
             dataset_name, name="summarized", split="train", streaming=True, token=oauth_token.token
-        ).select_columns(["raw_document_summary", "document_summary", "summarization_model"])
+        ).select_columns(["document_summary", "summarization_model"])
         summarization_df = pd.DataFrame(summarization_ds.take(5))
 
     if STAGE_DISPLAY_MAP["single_shot_question_generation"] in stages: