Update app.py

fix transparent background

app.py

@@ -24,8 +24,19 @@ def process_image(input_image, output_type):
         return None
     
     if output_type == "Transparent Background (RGBA)":
-        # Returns the masked image with transparent background
-        output = remover.process(input_image, type='rgba')
+        # Get the mask first
+        mask = remover.process(input_image, type='map')
+        
+        # Convert mask to proper format
+        if isinstance(mask, Image.Image):
+            mask = mask.convert('L')
+        else:
+            mask = Image.fromarray((mask * 255).astype(np.uint8), mode='L')
+        
+        # Convert original image to RGBA and apply mask as alpha channel
+        img_rgba = input_image.convert('RGBA')
+        img_rgba.putalpha(mask)
+        output = img_rgba
         
     elif output_type == "Green Background":
         # Returns the masked image with green screen background