from .db import init_db, get_db, Prediction