app.py

from fastapi import FastAPI
from dotenv import load_dotenv
from tasks import image

# Load environment variables
load_dotenv()

app = FastAPI(
    title="Frugal AI Challenge API",
    description="API for the Frugal AI Challenge evaluation endpoints"
)

# Include all routers
app.include_router(image.router)

@app.get("/")
async def root():
    return {
        "message": "This is an API for a frugal fire smoke detection model",
        "endpoints": {
            "image": "/image - Image classification task",
        }
    }