Add logging to the HF dataset

app.py

@@ -1,9 +1,13 @@
+import os
 import random
 import uuid
 
 import gradio as gr
 from datasets import load_dataset
 
+HF_TOKEN = os.environ.get('HF_TOKEN')
+HF_DATASET = os.environ.get('HF_DATASET')
+
 configuration = "commitchronicle-py-long"  # select a configuration
 dataset = load_dataset("JetBrains-Research/lca-cmg",
                        configuration,
@@ -11,7 +15,7 @@ dataset = load_dataset("JetBrains-Research/lca-cmg",
                        cache_dir="data")
 n_samples = len(dataset)
 
-saver = gr.CSVLogger()
+saver = gr.HuggingFaceDatasetSaver(HF_TOKEN, HF_DATASET)
 
 
 def get_github_link(repo, hash):