rockerritesh commited on
Commit
dbf4b37
·
verified ·
1 Parent(s): 438c548

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -168,7 +168,7 @@ async def batch_extract_text(background_tasks: BackgroundTasks, files: List[Uplo
168
  start_time = time.time()
169
 
170
  # Limit batch size for memory management
171
- max_batch_size = 5 # Adjust based on your GPU memory
172
 
173
  if len(files) > 32:
174
  return {"error": "A maximum of 20 images can be processed at a time."}
 
168
  start_time = time.time()
169
 
170
  # Limit batch size for memory management
171
+ max_batch_size = 20 # Adjust based on your GPU memory
172
 
173
  if len(files) > 32:
174
  return {"error": "A maximum of 20 images can be processed at a time."}