VeuReu commited on
Commit
0f7c8db
·
verified ·
1 Parent(s): a295921

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -0
api.py CHANGED
@@ -34,6 +34,7 @@ from main_process.main_router import router as main_router
34
  from main_process.salamandra_router import router as salamandra_router
35
  from main_process.moe_router import router as moe_router
36
  from main_process.refinement_router import router as refinement_router
 
37
 
38
  app = FastAPI(title="Veureu Engine API", version="0.2.0")
39
  app.add_middleware(
@@ -63,6 +64,7 @@ class JobStatus(str, Enum):
63
 
64
  jobs: Dict[str, dict] = {}
65
 
 
66
  app.include_router(main_router)
67
  app.include_router(salamandra_router)
68
  app.include_router(moe_router)
 
34
  from main_process.salamandra_router import router as salamandra_router
35
  from main_process.moe_router import router as moe_router
36
  from main_process.refinement_router import router as refinement_router
37
+ from storage.data_routers import router as data_router
38
 
39
  app = FastAPI(title="Veureu Engine API", version="0.2.0")
40
  app.add_middleware(
 
64
 
65
  jobs: Dict[str, dict] = {}
66
 
67
+ app.include_router(data_router)
68
  app.include_router(main_router)
69
  app.include_router(salamandra_router)
70
  app.include_router(moe_router)