Update test_api.py

test_api.py

@@ -72,7 +72,7 @@ def run_adain(content_dir, style_dataset_pth, out_dir, alpha=1.0, dataset_size=1
 
 		for idx in indices:
 			style_img = style_ds[idx]['image']
-			if style_img.mode not in ("RGB", "L"):
+			if not style_img.mode == "RGB":
 				style_img = style_img.convert("RGB")
 			style_tensor = t(style_img).unsqueeze(0).to(device)