Spaces:
Runtime error
Runtime error
File size: 779 Bytes
aae629d a7b721e aae629d 9fad749 6b24d30 9fad749 aae629d 6b24d30 9fad749 6b24d30 9fad749 1970c43 6b24d30 1970c43 6b24d30 a7b721e 6b24d30 1970c43 a7b721e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from fastapi import FastAPI
from .routes import router
import os
app = FastAPI(
title="Mila-Wellnest API",
description="Pregnancy Risk Prediction backend powered by fine-tuned DistilBERT.",
version="1.0.0"
)
@app.get("/")
def root():
return {"message": "👶 Hello from Mila-Wellnest Backend API"}
# Mount your prediction endpoint
app.include_router(router)
# Optional: Run training once on startup if model is missing
@app.on_event("startup")
async def startup_event():
checkpoint_path = "./data/best_model" # ✅ persistent path
if not os.path.exists(checkpoint_path):
os.makedirs(checkpoint_path, exist_ok=True)
print("⚙️ No checkpoint found. Starting training...")
from backend.app.train import train
train()
|