adithya9903's picture
Deploy PolyGuard HF training Space
fd0c71a verified
"""API application entrypoint."""
from __future__ import annotations
import os
import uvicorn
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from app.api.routes import router
app = FastAPI(title="POLYGUARD-RL API", version="0.1.0")
app.add_middleware(
CORSMiddleware,
allow_origins=[
"http://127.0.0.1:5173",
"http://localhost:5173",
],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
app.include_router(router)
def main() -> None:
host = os.getenv("POLYGUARD_API_HOST", "127.0.0.1")
port = int(os.getenv("POLYGUARD_API_PORT", "8200"))
uvicorn.run("app.api:app", host=host, port=port, reload=False)
if __name__ == "__main__":
main()