achdaisy commited on
Commit
05912ed
1 Parent(s): e8fa02f

Upload main.py

Browse files
Files changed (1) hide show
  1. main.py +31 -0
main.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI,Request
2
+ from fastapi.middleware.cors import CORSMiddleware
3
+
4
+ app = FastAPI()
5
+
6
+
7
+ app.add_middleware(
8
+ CORSMiddleware,
9
+ allow_origins=["*"],
10
+ allow_credentials=True,
11
+ allow_methods=["*"],
12
+ allow_headers=["*"],
13
+ )
14
+
15
+ @app.get("/ping")
16
+ async def ping():
17
+ return "Hello, the retriever is working"
18
+
19
+
20
+ @app.post("/qa")
21
+ async def section(request: Request):
22
+
23
+ data = await request.json()
24
+
25
+ from book_metadata_retriever import answer
26
+ answer = answer(data["Question"])
27
+
28
+ return {"Answer": answer}
29
+
30
+
31
+