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)