code: def from_imdb(
      batch_size: int = 4,
      **data_module_kwargs,
  ) -> TextClassificationData:
      """Downloads and loads the IMDB sentiment classification data set."""
      download_data(
          "https://pl-flash-data.s3.amazonaws.com/imdb.zip", "./data/"
      )
      return TextClassificationData.from_csv(
          "review",
          "sentiment",
          train_file="data/imdb/train.csv",
          val_file="data/imdb/valid.csv",
          batch_size=batch_size,
          **data_module_kwargs,
      )
  
imports: from flash.core.data.utils import download_data
  from flash.core.utilities.flash_cli import FlashCLI
  from flash.text import TextClassificationData, TextClassifier
label_desc: Use image node_id 1 for calling a global function with example usage: from_imdb(batch_size) and returns: TextClassificationData