The following is a repository for the final project for Transfromers in CV course.
Run the following command in your virtual environment to install Poetry using the official installer:
pipx install poetry
Set your local configuration with:
poetry config virtualenvs.create false --local
Install all necessary dependencies from pyproject.toml:
poetry install
- Download additional datasets using
get_dataset.shif you wish. - Run
eval_embeddings.pyto update embeddings.python3 eval_embeddings.py - Run
main.pyto generate memesUse flagpython3 main.py-pto pass your own meme caption, flag-tto pass a topic for a caption, and flag-dto specify path to the dataset.
- Boris Mikheev
- Ivan Listopadov
- Roman Makarov
- Sergey Grozny
- Yurii Melnik