README.md

---
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