Update main.py

main.py

@@ -162,13 +162,13 @@ async def extract_text(background_tasks: BackgroundTasks, file: UploadFile = Fil
         return {"error": str(e)}
 
 @app.post("/batch_extract_text")
-async def batch_extract_text(background_tasks: BackgroundTasks, files: List[UploadFile] = File(...)):
+async def batch_extract_text(batch_size:int, background_tasks: BackgroundTasks, files: List[UploadFile] = File(...)):
     """Extract text from multiple images with batching"""
     try:
         start_time = time.time()
         
         # Limit batch size for memory management
-        max_batch_size = 7  # Adjust based on your GPU memory
+        max_batch_size = batch_size  # Adjust based on your GPU memory
         
         # if len(files) > 32:
         #     return {"error": "A maximum of 20 images can be processed at a time."}