| 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) |