Updated image to validation

notebooks/template-image.ipynb

@@ -213,7 +213,7 @@
     "\n",
     "# Split dataset\n",
     "train_test = dataset[\"train\"]\n",
-    "test_dataset = dataset[\"test\"]"
+    "test_dataset = dataset[\"val\"]"
    ]
   },
   {

tasks/image.py

@@ -89,7 +89,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     
     # Split dataset
     train_test = dataset["train"]
-    test_dataset = dataset["test"]
+    test_dataset = dataset["val"]
     
     # Start tracking emissions
     tracker.start()