data/load_dataset.py

import logging
from datasets import load_dataset

def load_data():
    logging.info("Loading dataset")
    dataset = load_dataset("rungalileo/ragbench", 'covidqa', split="train")
    logging.info("Dataset loaded successfully")
    return dataset