Update tasks/audio.py

tasks/audio.py

@@ -37,8 +37,10 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     LABEL_MAPPING = {"chainsaw": 0, "environment": 1}
     # Load and prepare the dataset
     # Because the dataset is gated, we need to use the HF_TOKEN environment variable to authenticate
-    dataset = load_dataset(request.dataset_name, token=os.getenv("HF_TOKEN"))
-
+    print("Begin to load data...")
+    dataset = load_dataset(request.dataset_name, token=os.getenv("read-token"))
+    print("Data Loaded !")
+    
     # Split dataset
     train_test = dataset["train"].train_test_split(
         test_size=request.test_size, seed=request.test_seed