Estherrr777's picture
Update README.md
c825113 verified
metadata
title: Mila Wellnest Backend
emoji: 🀰
colorFrom: pink
colorTo: red
sdk: docker
pinned: true
license: apache-2.0
short_description: AI support for pregnancy health and mental wellness

Mila Wellnest 🀰✨

Mila Wellnest is an AI4Good project designed to support pregnant individuals by providing:

  • βœ… Risk level prediction based on clinical metrics (e.g., blood pressure, blood sugar, temperature)
  • βœ… Community-centered support for mental health (including postpartum depression screening)
  • βœ… Dynamic backend (FastAPI) and frontend (Gradio) architecture
  • βœ… Hugging Face–hosted, GPU-friendly, open-source deployment

πŸ”§ Tech Stack

  • Model: Fine-tuned Gemma or DistilBERT (transformers)
  • Backend: FastAPI (Docker)
  • Frontend: Gradio (in separate Space)
  • Infrastructure: Hugging Face Spaces

πŸ“¦ Architecture

Mila-Wellnest/
β”œβ”€β”€ backend/
β”‚   β”œβ”€β”€ app/                # FastAPI app, model, training scripts
β”‚   β”œβ”€β”€ data/               # JSON training data, saved checkpoints
β”‚   └── requirements.txt
β”œβ”€β”€ Dockerfile              # Launch FastAPI on port 7860