Update app.py

app.py

@@ -272,9 +272,11 @@ def infer(
     # If it's a single image (not a list), convert to list
     if not isinstance(input_images, list):
         input_images = [input_images]
+
+    images = [img[0] for img in input_images]
     
     # Concatenate images horizontally
-    concatenated_image = concatenate_images(input_images, "horizontal")
+    concatenated_image = concatenate_images(images, "horizontal")
     
     if concatenated_image is None:
         raise gr.Error("Failed to process the input images.")