This repo is meant as a template for projects using FastAPI and Streamlit for ML projects. A model is trained with the Bank Notes dataset in order to predict whether a Bank Note is fake or not.
You can create the environment to test out the project by simply running the following commands in the root directory:
-
make conda-update
to add exact Python and CUDA versions. -
conda activate fastapi-streamlit-ml
to create conda environment. -
make poetry
to install all Python dependencies. -
Create a .env file in the root directory with the following environment variables:
DB_USER DB_PASSWORD DB_HOST DB_PORT DB_NAME
-
make run-db
to create our PostgreSQL database with Docker