Fix Notebook?

[RED3480]

Can anyone help me put this into a streamlined notebook? I just want to input photos and output descriptions based on the pre-trained models. https://colab.research.google.com/drive/13IfMWEj1bEqCsyQK64qKnPyloB5lFfZ_?usp=sharing
I'm getting several errors. Thanks.

An assertion error in step 2 - If I use official data release instead of preprocess I don't get this error. Maybe due to the 14468 vs 14469 discrepency?

assert each image has at least 5 (human) votes!
x = image_distibutions.apply(sum)
assert all(x.values >= 5)

A split error in step 3
prepare data
data_loaders, datasets = image_emotion_distribution_df_to_pytorch_dataset(artemis_data, args)

An attribute error with sample speaker
!python '/content/artemis/artemis/scripts/sample_speaker.py'
-speaker-saved-args '/content/config.json.txt'
-speaker-checkpoint '/content/best_model.pt'
-img-dir '/content/Input'
-out-file .'/content/Output'
--custom-data-csv '/content/artemis_preprocessed.csv'

And a interpolation warning from nearest neighbor
Extract features
device = torch.device("cuda:" + gpu_id)
train_feats = extract_visual_features(train_images, img_dim, method=method, device=device)
test_feats = extract_visual_features(test_images, img_dim, method=method, device=device)

UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum.
"Argument interpolation should be of type InterpolationMode instead of int. "

[yiren-jian]

It's very important to follow the instruction to make a clean environment. I was using nltk==3.2.5 which give me 14468, after updating to nltk==3.4.5 (as suggested in setup.py), it's 14469 now.