Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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 =
|
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."}
|