Update app.py

app.py

@@ -6,13 +6,16 @@ from transformers import AutoModelForImageSegmentation
 import torch
 from torchvision import transforms
 
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 torch.set_float32_matmul_precision("high")
 
+
 model = AutoModelForImageSegmentation.from_pretrained(
     "ZhengPeng7/BiRefNet", trust_remote_code=True
 ).to(device)
 
+
 transform_image = transforms.Compose(
     [
         transforms.Resize((1024, 1024)),
@@ -26,27 +29,32 @@ def remove_background(image):
     im = load_img(image, output_type="pil").convert("RGB")
     image_size = im.size
 
+
     input_image = transform_image(im).unsqueeze(0).to(device)
     with torch.no_grad():
         preds = model(input_image)[-1].sigmoid().cpu()
-    
+
+
     mask = transforms.ToPILImage()(preds[0].squeeze()).resize(image_size)
-    im.putalpha(mask)
-    
-    return im, im.copy()
+    im_with_alpha = im.copy()
+    im_with_alpha.putalpha(mask)
 
-slider1 = ImageSlider(label="Arka Plan Kaldırıldı", type="pil")
-slider2 = ImageSlider(label="Arka Plan Kaldırıldı", type="pil")
+    return [im_with_alpha, im]
+
+
+slider = ImageSlider(label="Arka Plan Kaldırıldı", images=[""])
 image_input = gr.Image(label="Resim Yükle")
 url_input = gr.Textbox(label="Resim URL'si")
 
+
 chameleon = load_img("elon.webp", output_type="pil")
 url_example = "https://dadanizm.com/wp-content/uploads/2022/11/elon-musk-992x2656.jpeg"
 
+
 tab1 = gr.Interface(
     fn=remove_background, 
     inputs=image_input, 
-    outputs=slider1, 
+    outputs=slider, 
     examples=[chameleon], 
     api_name="image"
 )
@@ -54,11 +62,12 @@ tab1 = gr.Interface(
 tab2 = gr.Interface(
     fn=remove_background, 
     inputs=url_input, 
-    outputs=slider2, 
+    outputs=slider, 
     examples=[url_example], 
     api_name="url"
 )
 
+
 demo = gr.TabbedInterface(
     [tab1, tab2], 
     ["Resim", "URL"], 
@@ -66,4 +75,4 @@ demo = gr.TabbedInterface(
 )
 
 if __name__ == "__main__":
-    demo.launch(show_error=True)
+    demo.launch(show_error=True)