from datasets import load_from_disk
dataset_dict = load_from_disk("gs://medical-siddharth/medical_dataset")
dataset_dict.save_to_disk("medical_dataset")