refoundd commited on
Commit
72c4ebd
·
verified ·
1 Parent(s): 9b26d4e

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -0
handler.py CHANGED
@@ -216,6 +216,8 @@ class EndpointHandler:
216
  " prompt to use for the image generation, and it needs to be a non-empty string."
217
  )
218
  start_index,limit_count=prompt.split(',')
 
 
219
  logger.info(f"Start index: {start_index}, Limit count: {limit_count}")
220
  data = list(self.collection.find().skip(start_index).limit(limit_count))
221
  start_time=time.time()
 
216
  " prompt to use for the image generation, and it needs to be a non-empty string."
217
  )
218
  start_index,limit_count=prompt.split(',')
219
+ start_index=int(start_index)
220
+ limit_count=int(limit_count)
221
  logger.info(f"Start index: {start_index}, Limit count: {limit_count}")
222
  data = list(self.collection.find().skip(start_index).limit(limit_count))
223
  start_time=time.time()