sparrow-API

Runtime error

sparrow-API / endpoints.py

Sparrow ML new services

dfcd6b0 about 1 year ago

No virus

722 Bytes

	from fastapi import FastAPI
	from fastapi.middleware.cors import CORSMiddleware
	from routers import inference, training
	from huggingface_hub import login
	from config import settings

	login(settings.huggingface_key)

	app = FastAPI(openapi_url="/api/v1/sparrow-ml/openapi.json", docs_url="/api/v1/sparrow-ml/docs")

	app.add_middleware(
	CORSMiddleware,
	allow_origins=["*"],
	allow_methods=["*"],
	allow_headers=["*"],
	allow_credentials=True,
	)

	app.include_router(inference.router, prefix="/api-inference/v1/sparrow-ml", tags=["Inference"])
	app.include_router(training.router, prefix="/api-training/v1/sparrow-ml", tags=["Training"])


	@app.get("/")
	async def root():
	return {"message": "Sparrow ML API"}