fix syntax in yolov8 (#1790)

cc @paularamo
openvinotoolkit · Mar 6, 2024 · d604a22 · d604a22
1 parent e7a8b27
commit d604a22
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/notebooks/230-yolov8-optimization/230-yolov8-object-detection.ipynb b/notebooks/230-yolov8-optimization/230-yolov8-object-detection.ipynb
@@ -1151,9 +1151,10 @@
    "source": [
     "if device.value != \"CPU\":\n",
     "    quantized_det_model.reshape({0: [1, 3, 640, 640]})\n",
+    "ov_config = {}\n",
     "if \"GPU\" in device.value or (\"AUTO\" in device.value and \"GPU\" in core.available_devices):\n",
     "    ov_config = {\"GPU_DISABLE_WINOGRAD_CONVOLUTION\": \"YES\"}\n",
-    "quantized_det_compiled_model = core.compile_model(quantized_det_model, device.value, ov_config {\"GPU_DISABLE_WINOGRAD_CONVOLUTION\": \"YES\"} )\n",
+    "quantized_det_compiled_model = core.compile_model(quantized_det_model, device.value, ov_config)\n",
     "input_image = np.array(Image.open(IMAGE_PATH))\n",
     "detections = detect(input_image, quantized_det_compiled_model)[0]\n",
     "image_with_boxes = draw_results(detections, input_image, label_map)\n",