NLP-RAG / backend /api.py
Qar-Raz's picture
hf-space: deploy branch without frontend/data/results
c7256ee
raw
history blame contribute delete
942 Bytes
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from backend.routes.health import router as health_router
from backend.routes.predict import router as predict_router
from backend.routes.predict_stream import router as predict_stream_router
from backend.routes.title import router as title_router
from backend.services.startup import initialize_runtime_state
from backend.state import state
# fastapi configs defined here
# all the router objects are imported here
#--@Qamar
app = FastAPI(title="RAG-AS3 API", version="0.1.0")
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
app.include_router(health_router)
app.include_router(title_router)
app.include_router(predict_router)
app.include_router(predict_stream_router)
@app.on_event("startup")
def startup_event() -> None:
initialize_runtime_state(state)