Update tasks/audio.py

tasks/audio.py

@@ -43,21 +43,11 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     }
     # 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("RYmV25"))
-
-    request = AudioEvaluationRequest()
-    
-    # Load and prepare the dataset
-    dataset = load_dataset(request.dataset_name)
-    
-    # Split dataset
-    train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
-    test_dataset = train_test["test"]
-
-    
+    dataset = load_dataset(request.dataset_name,token=os.getenv("HF_TOKEN"))
+  
     # Split dataset
-    #train_test = dataset["train"]
-    #test_dataset = dataset["test"]
+    train_test = dataset["train"]
+    test_dataset = dataset["test"]
     
     # Start tracking emissions
     tracker.start()