code: def from_coco_128(
      batch_size: int = 4,
      num_workers: int = 0,
      **data_module_kwargs,
  ) -> StyleTransferData:
      """Downloads and loads the COCO 128 data set."""
      download_data(
          "https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.3.0/coco128.zip",
          "data/",
      )
      return StyleTransferData.from_folders(
          train_folder="data/coco128/images/train2017/",
          batch_size=batch_size,
          num_workers=num_workers,
          **data_module_kwargs,
      )
  
imports: import os
  import flash
  from flash.core.data.utils import download_data
  from flash.core.utilities.flash_cli import FlashCLI
  from flash.image import StyleTransfer, StyleTransferData
label_desc: Use image node_id 1 for calling a global function with example usage: from_coco_128(batch_size, num_workers) and returns: StyleTransferData