Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -170,8 +170,8 @@ async def batch_extract_text(background_tasks: BackgroundTasks, files: List[Uplo
|
|
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 |
-
|
175 |
|
176 |
load_model() # Ensure model is loaded
|
177 |
|
|
|
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."}
|
175 |
|
176 |
load_model() # Ensure model is loaded
|
177 |
|