fairchem
This commit is contained in:
7
main.py
7
main.py
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user