Update handler.py
Browse files- 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()
|