Update app.py

app.py

@@ -21,8 +21,6 @@ def detect_objects(images):
 def create_solutions(image_urls, names):
     solutions = []
     for image_url, prediction in zip(image_urls, names):
-        # prediction_list = []
-        # prediction_list.append(prediction)
         obj = {"url": image_url, "answer": [prediction]}
         solutions.append(obj)
     return solutions
@@ -48,13 +46,13 @@ def process_images(params):
         logging.error(f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}")
         return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
 
-    image_urls = params.get("image_urls", [])
+    image_urls = params.get("urls", [])
     # api = params.get("api", "")
     # job_id = params.get("job_id", "")
 
     if not image_urls:
-        logging.error("Missing required parameters: 'image_urls'")
-        return {"error": "Missing required parameters: 'image_urls'"}
+        logging.error("Missing required parameters: 'urls'")
+        return {"error": "Missing required parameters: 'urls'"}
 
     try:
         images = [Image.open(requests.get(url, stream=True).raw) for url in image_urls]
@@ -70,7 +68,7 @@ def process_images(params):
 
     return json.dumps({"solutions": solutions})
 
-inputt = gr.Textbox(label="Parameters (JSON format) Eg. {'image_urls':['a.jpg','b.jpg']}")
+inputt = gr.Textbox(label="Parameters (JSON format) Eg. {'urls':['a.jpg','b.jpg']}")
 outputs = gr.JSON()
 
 application = gr.Interface(fn=process_images, inputs=inputt, outputs=outputs, title="Car Colour Classification with API Integration")