Spaces:
Runtime error
Runtime error
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