Add wordlist

app.py

@@ -5,7 +5,11 @@ import random
 import numpy as np
 import torch
 from torchvision.transforms.functional import center_crop
+from datasets import load_dataset
 
+
+word_list_dataset = load_dataset("EPFL-VILAB/4m-wordlist", data_files="list.txt", use_auth_token=True)
+word_list = word_list_dataset["train"]['text']
 try:
     # Try to install detectron2 from source. Needed for semseg plotting functionality.
     os.system("python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'")