Add secret

app.py

@@ -1,8 +1,18 @@
+import os
 import gradio as gr
 import matplotlib.pyplot as plt
 import numpy as np
 from functools import partial
 from datasets import load_dataset
+from pathlib import Path
+
+# get secret environment variable
+token = os.environ["HF_TOKEN"]
+# write to disk
+path = "./huggingface"
+Path(path).mkdir(parents=True, exist_ok=True)
+with open(path/"token", "w") as f:
+    f.write(token)
 
 dataset_names = [
     "AI4Code",
@@ -67,6 +77,9 @@ def plt_plot(ratio, dataset, threshold):
     plt.tight_layout()
     return fig
 
+def check_filtered():
+    ...
+
 with gr.Blocks() as demo:
     dataset = gr.Radio(dataset_names, label="Dataset", value="arXiv")
     print(dataset.value)
@@ -76,6 +89,7 @@ with gr.Blocks() as demo:
         plot = gr.Plot()
         threshold = gr.Slider(minimum=0, maximum=1, label="Threshold")
         calculate = gr.Button("Calculate")
+        check = gr.Button("Check Filtered Data")
         plot_fn = partial(plt_plot, "char_rep_ratios")
         calculate.click(plot_fn, [dataset, threshold], plot)