A system that, when given a role description, returns a short list of actors who fit that role.
We achieved this by:
- Finding, Cleaning, and Processing data on actors and movies
- Utilizing BERT-based natural language processing model
- Using cosine similarity to find matches
- Implementing preference characteristics
- Simple UI for usability and testing
- Create your virtual environment
python3 -m venv .venv
- Activate your virtual environment
source .venv/bin/activate
- Download requirements (with virtual env active)
pip install -r requirements.txt
- Run the app
streamlit run src/actor_rec/main.py