Sirorororo commited on
Commit
783894a
·
verified ·
1 Parent(s): b94a4c6

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 = 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."}
 
168
  start_time = time.time()
169
 
170
  # Limit batch size for memory management
171
+ max_batch_size = 7 # 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."}