code: def __init__(
      self, data_source, column_names=["image"], label_column="label"
  ):
      self.data_source = data_source
      self.column_names = column_names
      self.label_column = label_column
  
imports: import torch
  import pandas
  from tqdm import tqdm
  import torch.nn.functional
  import torch.nn
  from autogluon.multimodal import MultiModalPredictor
  from datasets import list_datasets, load_dataset
  from setfit import sample_dataset
  from imagedatasets import build_dataset
  from torch.utils.data import Dataset
label_desc: Use image node_id 1 to create a new Wrapper object from inherited base classes: TorchDataset with example: obj = Wrapper(data_source, column_names, label_column)