add example image

app.py

@@ -5,17 +5,22 @@ from matplotlib import pyplot as plt
 import os, shutil
 
 import gradio as gr
-from PIL import Image
+from PIL import Image, ImageOps
 import sys, os
 from rembg import remove
 
 
 def run_rembg(img):
     output = remove(img)
-    return output
+    output_pil = Image.fromarray(output)
+
+    # Remove margins 
+    cropped_image = output_pil.crop(output_pil.getbbox())
+
+    return cropped_image
 
 def load_zip(inputs):
-    os.makedirs("results", exist_ok=True)
+    os.makedirs("objects", exist_ok=True)
 
     image_data_dict = {}
     with ZipFile(inputs[0].name, "r") as zip_file:
@@ -46,18 +51,23 @@ def load_zip(inputs):
     
     
     for image_name, image_data in image_data_dict.items():
-        img_array = remove(image_data)        
+        output = remove(image_data)        
+        output_pil = Image.fromarray(output)
+        # Remove margins 
+        cropped_image = output_pil.crop(output_pil.getbbox())
 
         image_name = image_name.replace("jpg", "png")
-        plt.imsave(f"results/{image_name}", img_array)
+        cropped_image.save(f"objects/{image_name}")
+        # plt.imsave(f"objects/{image_name}", img_array)
     
-    shutil.make_archive("results", "zip", "results")
-    with ZipFile("objects.zip", "w") as zip_file:
-        zip_file.write("results.zip")
-
-    shutil.rmtree("results")
+    shutil.make_archive("objects", "zip", "objects")
+    # ここはresults
+    with ZipFile("results.zip", "w") as zip_file:
+        zip_file.write("objects.zip")
 
-    return "objects.zip"
+    shutil.rmtree("objects")
+    # ここはresults
+    return "results.zip"
 
 
 
@@ -73,21 +83,22 @@ if __name__=="__main__":
             """
             )        
             with gr.Row():
-                with gr.Column():
-                    gr.Markdown(
-                        """
-                        Input Zip File
-                        """
-                    )        
-                    image_input = gr.File(file_count="multiple")
-                
-                with gr.Column():
-                    gr.Markdown(
-                        """
-                        Output Zip File
-                        """
-                    )        
-                    image_output = gr.File()
+                gr.Markdown(
+                    """
+                    ### Input Zip File
+                    <img src='file/assets/input_zip.png' width="85%" height="85%">
+                    """
+                )        
+                gr.Markdown(
+                    """
+                    ### Output Zip File
+                    <img src='file/assets/output_zip.png' width="85%" height="85%">
+                    """
+                )        
+            
+            with gr.Row():
+                image_input = gr.File(file_count="multiple")
+                image_output = gr.File()
             
             btn = gr.Button("Run!")
             
@@ -105,20 +116,22 @@ if __name__=="__main__":
                 """
             )        
             with gr.Row():
-                with gr.Column():
-                    gr.Markdown(
-                        """
-                        Input Image
-                        """
-                    )        
-                    image_input = gr.Image(type="numpy")
-                with gr.Column():
-                    gr.Markdown(
-                        """
-                        Output Image
-                        """
-                    )        
-                    image_output = gr.Image(type="numpy")
+                gr.Markdown(
+                    """
+                    ### Input Image
+                    <img src='file/assets/cat-3038243_1280.jpg' width="50%" height="50%">
+                    """
+                )        
+                
+                gr.Markdown(
+                    """
+                    ### Output Image
+                    <img src='file/assets/objects/cat-3038243_1280.png' width="20%" height="20%">
+                    """
+                )        
+            with gr.Row():
+                image_input = gr.Image(type="numpy")
+                image_output = gr.Image(type="pil")
             
             btn = gr.Button("Run!")
         
@@ -138,7 +151,8 @@ if __name__=="__main__":
               - Cloned on 2023/3/12  
             - Algorithm        
               - Library Git hub : [U<sup>2</sup>-Net](https://github.com/xuebinqin/U-2-Net)  
-            
+            - Image
+              - Cat Image from [Pixabay](https://pixabay.com/images/id-3038243/)
             """
         )