Upload folder using huggingface_hub

.gitattributes

@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 example1.png filter=lfs diff=lfs merge=lfs -text
 example2.png filter=lfs diff=lfs merge=lfs -text
 example3.png filter=lfs diff=lfs merge=lfs -text
+examples.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -71,7 +71,14 @@ description = r"""
 This model is a <strong>fully open-source background remover</strong> optimized for images with humans.
 If you identify cases were the model fails, <a href='https://huggingface.co/schirrmacher/ormbg/discussions' target='_blank'>please contact me</a>!
 
-<a href='https://huggingface.co/schirrmacher/ormbg' target='_blank'>Model information here.</a>
+- <a href='https://huggingface.co/schirrmacher/ormbg' target='_blank'>Model card: inference code</a>
+- <a href='https://huggingface.co/schirrmacher/ormbg' target='_blank'>Dataset: all images and backgrounds</a>
+
+Known issues (work in progress):
+- close-ups: from above, from below, profile, from side
+- minor issues with hair segmentation when hair creates loops
+- more various backgrounds needed
+
 """
 examples = ["./example1.png", "./example2.png", "./example3.png"]
 

stack.py

@@ -0,0 +1,37 @@
+from PIL import Image
+
+
+def stack_images(image_paths, output_path):
+    # Load all images from the provided paths
+    images = [Image.open(path) for path in image_paths]
+
+    # Determine the size of individual images (assuming all are the same size)
+    width, height = images[0].size
+
+    # Create a new image with appropriate size (2 columns and 3 rows)
+    total_width = width * 2
+    total_height = height * 3
+    new_image = Image.new("RGB", (total_width, total_height))
+
+    # Paste each image into the new image
+    for i, image in enumerate(images):
+        # Calculate the position for each image
+        x_offset = (i % 2) * width
+        y_offset = (i // 2) * height
+        new_image.paste(image, (x_offset, y_offset))
+
+    # Save the new image
+    new_image.save(output_path)
+
+
+# Example usage
+image_paths = [
+    "/Users/mav/Desktop/example1.png",
+    "/Users/mav/Desktop/image-1.webp",
+    "/Users/mav/Desktop/example2.png",
+    "/Users/mav/Desktop/image-2.webp",
+    "/Users/mav/Desktop/example3.png",
+    "/Users/mav/Desktop/image-3.webp",
+]
+output_path = "stacked_images.jpg"
+stack_images(image_paths, output_path)