achdaisy's picture
Update main.py
e5b5d74 verified
raw
history blame contribute delete
539 Bytes
from fastapi import FastAPI,Request
from fastapi.middleware.cors import CORSMiddleware
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/ping")
async def ping():
return "Hello, the retriever is working"
@app.post("/qa")
async def section(request: Request):
data = await request.json()
from final_book_retriever import answer
answer = answer(data["Question"])
return {"Answer": answer}