Update tasks/audio.py

tasks/audio.py

@@ -37,9 +37,7 @@ 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
-    print("Begin to load data...")
-    dataset = load_dataset(request.dataset_name, token=os.getenv("read-token"))
-    print("Data Loaded !")
+    dataset = load_dataset(request.dataset_name,token=os.getenv("HF_TOKEN"))
     
     # Split dataset
     train_test = dataset["train"].train_test_split(