banao-tech commited on
Commit
c0e00b1
·
verified ·
1 Parent(s): abef7a7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -53,8 +53,8 @@ logger.info("Finished loading models!!!")
53
  # Initialize FastAPI app
54
  app = FastAPI()
55
 
56
- # Define a queue for request processing
57
- request_queue = asyncio.Queue()
58
 
59
  # Define a response model for the processed image
60
  class ProcessResponse(BaseModel):
 
53
  # Initialize FastAPI app
54
  app = FastAPI()
55
 
56
+ MAX_QUEUE_SIZE = 10 # Set a reasonable limit based on your system capacity
57
+ request_queue = asyncio.Queue(maxsize=MAX_QUEUE_SIZE)
58
 
59
  # Define a response model for the processed image
60
  class ProcessResponse(BaseModel):