This commit is contained in:
2025-01-05 17:43:06 +08:00
parent f214f51e12
commit 5380ee5f9e
4 changed files with 169 additions and 64 deletions

View File

@@ -8,6 +8,7 @@ from fastapi import FastAPI
import logging
from database.material_project_router import router as material_project_router
from database.oqmd_router import router as oqmd_router
from model.fairchem_router import router as fairchem_router, init_model
logging.basicConfig(
level=logging.INFO,
@@ -16,5 +17,11 @@ logging.basicConfig(
logger = logging.getLogger(__name__)
app = FastAPI()
@app.on_event("startup")
def startup_event():
init_model()
app.include_router(material_project_router)
app.include_router(oqmd_router)
app.include_router(fairchem_router)