Update app.py

app.py

@@ -6,16 +6,12 @@ from transformers import AutoModelForImageSegmentation
 import torch
 from torchvision import transforms
 
+torch.set_float32_matmul_precision(["high", "highest"][0])
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-torch.set_float32_matmul_precision("high")
-
-
-model = AutoModelForImageSegmentation.from_pretrained(
+birefnet = AutoModelForImageSegmentation.from_pretrained(
     "ZhengPeng7/BiRefNet", trust_remote_code=True
-).to(device)
-
-
+)
+birefnet.to("cpu")
 transform_image = transforms.Compose(
     [
         transforms.Resize((1024, 1024)),
@@ -24,55 +20,42 @@ transform_image = transforms.Compose(
     ]
 )
 
-@spaces.GPU
-def remove_background(image):
-    im = load_img(image, output_type="pil").convert("RGB")
-    image_size = im.size
-
+def fn(image):
+    im = load_img(image, output_type="pil")
+    im = im.convert("RGB")
+    origin = im.copy()
+    image = process(im)
+    image = image.resize(origin.size) 
+    return (image, origin)
 
-    input_image = transform_image(im).unsqueeze(0).to(device)
+@spaces.GPU
+def process(image):
+    input_images = transform_image(image).unsqueeze(0).to("cpu")
     with torch.no_grad():
-        preds = model(input_image)[-1].sigmoid().cpu()
-
-
-    mask = transforms.ToPILImage()(preds[0].squeeze()).resize(image_size)
-    im_with_alpha = im.copy()
-    im_with_alpha.putalpha(mask)
-
-    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")
-
+        preds = birefnet(input_images)[-1].sigmoid().cpu()
+    pred = preds[0].squeeze()
+    pred_pil = transforms.ToPILImage()(pred)
+    mask = pred_pil.resize(image.size)
+    image.putalpha(mask)
+    return image
+  
+slider1 = ImageSlider(label="Birefnet", type="pil")
+slider2 = ImageSlider(label="Birefnet", type="pil")
+image = gr.Image(label="Bir resim yükleyin")
+text = gr.Textbox(label="Bir resim URL'si yapıştırın")
 
 chameleon = load_img("elon.webp", output_type="pil")
-url_example = "https://dadanizm.com/wp-content/uploads/2022/11/elon-musk-992x2656.jpeg"
-
 
+url= "https://dadanizm.com/wp-content/uploads/2022/11/elon-musk-992x2656.jpeg"
 tab1 = gr.Interface(
-    fn=remove_background, 
-    inputs=image_input, 
-    outputs=slider, 
-    examples=[chameleon], 
-    api_name="image"
-)
-
-tab2 = gr.Interface(
-    fn=remove_background, 
-    inputs=url_input, 
-    outputs=slider, 
-    examples=[url_example], 
-    api_name="url"
+    fn, inputs=image, outputs=slider1, examples=[chameleon], api_name="image"
 )
 
+tab2 = gr.Interface(fn, inputs=text, outputs=slider2, examples=[url], api_name="text")
 
 demo = gr.TabbedInterface(
-    [tab1, tab2], 
-    ["Resim", "URL"], 
-    title="Resim Arka Planını Kaldır"
+    [tab1, tab2], ["Resim", "URL"], title="Birefnet ile Arka Plan Kaldırma"
 )
 
 if __name__ == "__main__":
-    demo.launch(show_error=True)
+    demo.launch(show_error=True)