Estherrr777's picture
Update README.md
c825113 verified
---
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
```bash
Mila-Wellnest/
β”œβ”€β”€ backend/
β”‚ β”œβ”€β”€ app/ # FastAPI app, model, training scripts
β”‚ β”œβ”€β”€ data/ # JSON training data, saved checkpoints
β”‚ └── requirements.txt
β”œβ”€β”€ Dockerfile # Launch FastAPI on port 7860