Update tasks/audio.py

tasks/audio.py

@@ -43,11 +43,21 @@ 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"))
+    #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"]
+
     
     # Split dataset
-    train_test = dataset["train"]
-    test_dataset = dataset["test"]
+    #train_test = dataset["train"]
+    #test_dataset = dataset["test"]
     
     # Start tracking emissions
     tracker.start()