Use the authentication token

app.py

@@ -2,6 +2,7 @@
 # TODO: Secrets?
 # TODO: Dataset_name to Secrets and not hard-coded!
 
+import os
 import streamlit as st
 
 import torch
@@ -55,7 +56,7 @@ def predict_top_p(text, P=0.9):
 
 
 dataset_name = "AMR-KELEG/test-dataset"
-dataset = datasets.load_dataset(dataset_name)["test"]
+dataset = datasets.load_dataset(dataset_name, token=os.environ["HF_TOKEN"])["test"]
 
 for sample in dataset:
     text = sample["sentence"]