Update main.py

main.py

@@ -7,7 +7,6 @@ import requests
 import random
 from datetime import datetime
 import os
-from rembg import remove
 
 
 prodia_key = os.getenv('PRODIA_X_KEY', None)
@@ -46,9 +45,6 @@ def process_input_img2img(init, prompt, negative_prompt, steps, cfg_scale, numbe
                 f.write(img_data)
     return images
 
-def segment(image):
-     return remove(image)
-
 """
 def process_input_control(init, prompt, negative_prompt, steps, cfg_scale, number, seed, model, control_model, sampler):
     images = []
@@ -208,29 +204,9 @@ with gr.Blocks(theme=theme) as demo:
             outputs=[result_image],
         )
 
-    with gr.Tab("Remove Background"):
-        with gr.Row():
-        
-            with gr.Column():
-                i_img = gr.Image(label="Input Image")
-
-        with gr.Row():
-
-            with gr.Column():
-                run_btn = gr.Button("Remove", variant="primary")
-    
-        with gr.Row():
+    with gr.Tab(label="Remove Background"):
 
-            with gr.Column():
-                o_img = gr.Image(label="Result Image")
-
-        run_btn.click(
-                segment,
-                inputs=[
-                    i_img,
-                ],
-                outputs=[o_img],
-        )
+        gr.load("hardon-server/remove-background-on-image-def", src="spaces")
     
     with gr.Tab(label="Gallery"):