ImenMourali commited on
Commit
4768d6b
·
verified ·
1 Parent(s): d87ee35

Update tasks/audio.py

Browse files
Files changed (1) hide show
  1. tasks/audio.py +2 -2
tasks/audio.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  from .utils.evaluation import AudioEvaluationRequest
9
  from .utils.emissions import tracker, clean_emissions_data, get_space_info
10
 
11
- from dotenv import load_dotenv
12
  load_dotenv()
13
 
14
  router = APIRouter()
@@ -43,7 +43,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
43
  }
44
  # Load and prepare the dataset
45
  # Because the dataset is gated, we need to use the HF_TOKEN environment variable to authenticate
46
- dataset = load_dataset(request.dataset_name,token=os.getenv("HF_TOKEN"))
47
 
48
  # Split dataset
49
  train_test = dataset["train"]
 
8
  from .utils.evaluation import AudioEvaluationRequest
9
  from .utils.emissions import tracker, clean_emissions_data, get_space_info
10
 
11
+ from dotenv import load_dotenv
12
  load_dotenv()
13
 
14
  router = APIRouter()
 
43
  }
44
  # Load and prepare the dataset
45
  # Because the dataset is gated, we need to use the HF_TOKEN environment variable to authenticate
46
+ dataset = load_dataset(request.dataset_name,token=os.getenv("RYmV25"))
47
 
48
  # Split dataset
49
  train_test = dataset["train"]