Sugiro estudar e implementar em FastApi. Django é bem mais completo e complexo, talvez não seja sua necessidade no momento. Interessante seu projeto. Bacana.

Obrigado pela orientação.

Poderia me opinar sobre essa implementação? Ela ainda não está completa. https://github.com/gabriel-roque/neural-network-diabetes/blob/master/app.py