Update app.py

app.py

@@ -27,7 +27,7 @@ def fn(image: Union[Image.Image, str]) -> Tuple[Image.Image, Image.Image]:
     Remove the background from an image and return both the transparent version and the original.
 
     This function performs background removal using a BiRefNet segmentation model. It is intended for use
-    with image input (either uploaded or from a URL). The function returns a transparent PNG version of the image
+    with image input (either uploaded or from a URL). The function returns a transparent WEBP version of the image
     with the background removed, along with the original RGB version for comparison.
 
     Args:
@@ -76,27 +76,27 @@ def process(image: Image.Image) -> Image.Image:
 
 def process_file(f: str) -> str:
     """
-    Load an image file from disk, remove the background, and save the output as a transparent PNG.
+    Load an image file from disk, remove the background, and save the output as a transparent WEBP.
 
     Args:
         f (str): Filepath of the image to process.
 
     Returns:
-        str: Path to the saved PNG image with background removed.
+        str: Path to the saved WEBP image with background removed.
     """
-    name_path = f.rsplit(".", 1)[0] + ".png"
+    name_path = f.rsplit(".", 1)[0] + ".webp"
     im = load_img(f, output_type="pil")
     im = im.convert("RGB")
     transparent = process(im)
     transparent.save(name_path)
     return name_path
 
-slider1 = gr.ImageSlider(label="Processed Image", type="pil", format="png")
-slider2 = gr.ImageSlider(label="Processed Image from URL", type="pil", format="png")
+slider1 = gr.ImageSlider(label="Processed Image", type="pil")
+slider2 = gr.ImageSlider(label="Processed Image from URL", type="pil"
 image_upload = gr.Image(label="Upload an image")
 image_file_upload = gr.Image(label="Upload an image", type="filepath")
 url_input = gr.Textbox(label="Paste an image URL")
-output_file = gr.File(label="Output PNG File")
+output_file = gr.File(label="Output WEBP File")
 
 # Example images
 chameleon = load_img("butterfly.jpg", output_type="pil")
@@ -104,7 +104,7 @@ url_example = "https://hips.hearstapps.com/hmg-prod/images/gettyimages-122989298
 
 tab1 = gr.Interface(fn, inputs=image_upload, outputs=slider1, examples=[chameleon], api_name="image")
 tab2 = gr.Interface(fn, inputs=url_input, outputs=slider2, examples=[url_example], api_name="text")
-tab3 = gr.Interface(process_file, inputs=image_file_upload, outputs=output_file, examples=["butterfly.jpg"], api_name="png")
+tab3 = gr.Interface(process_file, inputs=image_file_upload, outputs=output_file, examples=["butterfly.jpg"], api_name="webp")
 
 demo = gr.TabbedInterface(
     [tab1, tab2, tab3], ["Image Upload", "URL Input", "File Output"], title="Background Removal Tool"