Update test_api.py

test_api.py

@@ -58,7 +58,7 @@ def run_adain(content_dir, style_dataset_pth, out_dir, alpha=1.0, dataset_size=1
 	# Timer
 	times = []
 
-	style_ds = load_dataset(style_dataset_pth, split="train", token=os.getenv("PALEO_DATASET"))
+	style_ds = load_dataset(style_dataset_pth, split="train", token=os.getenv("PALEO_SECRET"))
 
 	if num_content_imgs * len(style_ds) > dataset_size:
 		num_style_per_content = int(np.ceil(dataset_size / num_content_imgs))