| import uvicorn | |
| from dotenv import load_dotenv | |
| from fastapi import FastAPI | |
| from fastapi.middleware.cors import CORSMiddleware | |
| from api.base_router import base_router | |
| import os | |
| os.environ['TRANSFORMERS_HOME'] = './.cache' | |
| if __name__ == '__main__': | |
| load_dotenv() | |
| uvicorn.run( | |
| 'app:app', | |
| reload=True, | |
| ) | |
| app = FastAPI( | |
| docs_url='/docs', | |
| ) | |
| app.include_router(base_router) | |
| app.add_middleware( | |
| CORSMiddleware, | |
| allow_origins=["*"], | |
| allow_credentials=True, | |
| allow_methods=["*"], | |
| allow_headers=["*"], | |
| ) | |