app.py

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from modules.routers import setup_routers
import torch

app = FastAPI()
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"]
)

print(f"Is CUDA available: {torch.cuda.is_available()}")

setup_routers(app)