The code should run as is with the following dependencies: pip install transformers datasets baal matplotlib tqdm