transfer_learning_demo

How to run it:

Create a virtual environment

python -m venv venv
source /venv/bin/activate

Install requirements

pip install -r requirements.txt

Train model

python train_model.py

Run Streamlit demo

docker compose up