Update app.py

app.py

@@ -16,6 +16,9 @@ import gradio as gr
 from huggingface_hub import hf_hub_download, snapshot_download
 from ip_adapter import IPAdapterXL
 from safetensors.torch import load_file
+# remove the background
+from tsr.system import TSR
+from tsr.utils import remove_background, resize_foreground
 
 snapshot_download(
     repo_id="h94/IP-Adapter", allow_patterns="sdxl_models/*", local_dir="."
@@ -100,8 +103,30 @@ def resize_img(
             np.array(input_image)
         )
         input_image = Image.fromarray(res)
+    # return input_image
+    return resized_image
+
+    #added
+
+def preprocess(resized_image, do_remove_background, foreground_ratio):
+    def fill_background(image):
+        image = np.array(image).astype(np.float32) / 255.0
+        image = image[:, :, :3] * image[:, :, 3:4] + (1 - image[:, :, 3:4]) * 0.5
+        image = Image.fromarray((image * 255.0).astype(np.uint8))
+        return image
+
+    if do_remove_background:
+        image = input_image.convert("RGB")
+        image = remove_background(image, rembg_session)
+        image = resize_foreground(image, foreground_ratio)
+        image = fill_background(image)
+    else:
+        image = input_image
+        if image.mode == "RGBA":
+            image = fill_background(image)
     return input_image
 
+    #/added
 
 examples = [
     [
@@ -138,7 +163,7 @@ def run_for_examples(style_image, source_image, prompt, scale, control_scale):
     )
 
 
-@spaces.GPU(enable_queue=True)
+@spaces.GPU
 def create_image(
     image_pil,
     input_image,
@@ -257,6 +282,7 @@ with block:
                 with gr.Row():
                     with gr.Column():
                         image_pil = gr.Image(label="Style Image", type="pil")
+                        processed_image = gr.Image(label="Preprocess uWu", interactive=False)
                     with gr.Column():
                         prompt = gr.Textbox(
                             label="Prompt",