Update app.py

app.py

@@ -30,26 +30,11 @@ def remove_background_rembg(input_path):
     img = Image.open(io.BytesIO(output_image)).convert("RGBA")
     return img
 
-# Initialize the pipeline globally for bria
-pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True, device=0)
-
 def remove_background_bria(input_path):
     print(f"Removing background using bria for image: {input_path}")
-    
-    # Buka gambar menggunakan PIL
-    image = Image.open(input_path).convert("RGBA")  # Pastikan format RGBA
-    
-    # Proses penghapusan latar belakang
-    result = pipe(image)  # Pastikan pipe menerima objek PIL
-    
-    # Pastikan Anda memeriksa struktur hasilnya
-    if isinstance(result, list) and len(result) > 0:
-        # Mengambil gambar dari hasil
-        output_image = result[0]['segmentation']  # Atur sesuai dengan struktur hasil dari pipeline
-        output_image = Image.fromarray(output_image)  # Konversi kembali ke PIL
-        return output_image
-    else:
-        raise ValueError("Invalid output from bria pipeline")
+    pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True, device=0)
+    pillow_image = pipe(input_path)
+    return pillow_image
 
 # Fungsi untuk memproses gambar menggunakan prompt
 def text_to_image(prompt):