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=["*"], | |
) | |