Update app.py

app.py

@@ -1,3 +1,4 @@
+import io
 import os
 import sys
 
@@ -46,12 +47,14 @@ def read_root():
     return RedirectResponse(url="/docs")
 
 @app.post("/index")
-def index(files: List[UploadFile] = File(...)):
+async def index(files: List[UploadFile] = File(...)):
     global ds, images
     images = []
     ds = []
     for file in files:
-        images.extend(convert_from_path(file))
+        content = await file.read()
+        pdf_image_list = convert_from_path(io.BytesIO(content))
+        images.extend(pdf_image_list)
     
     dataloader = DataLoader(
         images,
@@ -68,7 +71,7 @@ def index(files: List[UploadFile] = File(...)):
     return {"message": f"Uploaded and converted {len(images)} pages"}
 
 @app.post("/search")
-def search(query: str, k: int):
+async def search(query: str, k: int):
     qs = []
     with torch.no_grad():
         batch_query = process_queries(processor, [query], mock_image)