| from contextlib import asynccontextmanager |
| from fastapi import FastAPI |
| from fastapi.middleware.cors import CORSMiddleware |
|
|
|
|
| @asynccontextmanager |
| async def lifespan(app): |
| from TextGen.router import startup |
| await startup() |
| yield |
|
|
|
|
| app = FastAPI(title="Deploying FastAPI Apps on Huggingface", lifespan=lifespan) |
| app.add_middleware( |
| CORSMiddleware, |
| allow_origins=["https://jofthomas-everchanging-quest.static.hf.space"], |
| allow_credentials=True, |
| allow_methods=["*"], |
| allow_headers=["*"], |
| ) |
|
|
| from TextGen import router |
|
|
|
|
|
|
