Update handler.py

handler.py

@@ -1,3 +1,5 @@
+# .env
+# MongoDB -> mongodb+srv://jamie:qJiuKQpqhXMHGb74@cluster0.i5ujz.mongodb.net/
 import os
 from typing import Any, Dict
 from PIL import Image
@@ -208,14 +210,13 @@ class EndpointHandler:
 
         if "inputs" in data and isinstance(data["inputs"], str):
             prompt = data.pop("inputs")
-        elif "prompt" in data and isinstance(data["prompt"], str):
-            prompt = data.pop("prompt")
         else:
             raise ValueError(
                 "Provided input body must contain either the key `inputs` or `prompt` with the"
                 " prompt to use for the image generation, and it needs to be a non-empty string."
             )
         start_index,limit_count=prompt.split(',')
+        logger.info(f"Start index: {start_index}, Limit count: {limit_count}")
         data = list(self.collection.find().skip(start_index).limit(limit_count)) 
         start_time=time.time()
         for document in data: