I think you have to start with
imdb = load_dataset("imdb", "plain_text")
as the configuration is named “plain_text”.