Set Up

1. Follow all commands here: https://github.com/rom1504/clip-retrieval/blob/main/docs/laion5B_back.md
2. source .env/bin/activate

Installations

For the image captioning model:

pip install transformers pillow

How to Run

After cloning the repo, enter this in the command line:

python kaggle_load.py

Datasets

We linked the datasets we used in Section 4.2 of our final report, submitted to Gradescope.