add model

app.py

@@ -31,7 +31,15 @@ def run_rembg(img):
 
     return cropped_image
 
+def run_rembg_withmodel(img,model):
+    output = remove(img)
+    output_pil = Image.fromarray(output)
+
+    # Remove margins 
+    cropped_image = output_pil.crop(output_pil.getbbox())
 
+    return cropped_image
+    
 def from_zip(inputs):
     work_dir = get_random_name()
     os.makedirs(work_dir, exist_ok=True)
@@ -240,7 +248,42 @@ if __name__=="__main__":
                 outputs=image_output,
                 api_name="imageMatting"
                 )
-        
+                
+        with gr.Tab("ImageWithModel"):
+            gr.Markdown(
+                """
+                <center><h1>Image Matting using different models</h1></center>
+                """
+            )        
+            with gr.Row():
+                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():
+                with gr.Column():
+                    model_name = gr.Textbox(label="Model Name", value="", placeholder="u2net")
+                    image_input = gr.Image(type="numpy")                
+                image_output = gr.Image(type="pil")
+            
+            btn = gr.Button("Run!")
+
+
+            btn.click(
+                fn=run_rembg_withmodel,
+                inputs=[image_input, model_name]
+                outputs=image_output,
+                api_name="imageMattingWithModel"
+                )