| """Optional FastAPI microservice for Imager-1-Pro.""" | |
| from fastapi import FastAPI, File, UploadFile | |
| from fastapi.responses import JSONResponse | |
| from io import BytesIO | |
| from PIL import Image | |
| from .config import load_model, predict | |
| app = FastAPI(title="Imager-1-Pro", version="1.0.0") | |
| _model = None | |
| def get_model(): | |
| global _model | |
| if _model is None: | |
| _model = load_model() | |
| return _model | |
| def root(): | |
| return {"name": "imager-1-pro", "status": "ok"} | |
| def health(): | |
| return {"status": "healthy"} | |
| async def predict_endpoint(file: UploadFile = File(...)): | |
| image = Image.open(BytesIO(await file.read())) | |
| result = predict(image, model=get_model()) | |
| return JSONResponse(result) | |