Update tasks/audio.py
Browse files- 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("
|
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"]
|