VeuReu commited on
Commit
795a1cd
·
verified ·
1 Parent(s): fe0e2a2

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +4 -1
api.py CHANGED
@@ -32,6 +32,8 @@ from storage.embeddings_routers import router as embeddings_router
32
  from storage.pending_videos_routers import router as pending_videos_router
33
  from main_process.main_router import router as main_router
34
  from main_process.salamandra_router import router as salamandra_router
 
 
35
 
36
  app = FastAPI(title="Veureu Engine API", version="0.2.0")
37
  app.add_middleware(
@@ -61,9 +63,10 @@ class JobStatus(str, Enum):
61
 
62
  jobs: Dict[str, dict] = {}
63
 
64
-
65
  app.include_router(main_router)
66
  app.include_router(salamandra_router)
 
 
67
  app.include_router(media_router)
68
  app.include_router(db_router)
69
  app.include_router(embeddings_router)
 
32
  from storage.pending_videos_routers import router as pending_videos_router
33
  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
 
64
  jobs: Dict[str, dict] = {}
65
 
 
66
  app.include_router(main_router)
67
  app.include_router(salamandra_router)
68
+ app.include_router(moe_router)
69
+ app.include_router(refinement_router)
70
  app.include_router(media_router)
71
  app.include_router(db_router)
72
  app.include_router(embeddings_router)