Update app.py

app.py

@@ -90,13 +90,11 @@ def transparent(fg, bg, alpha_factor):
 
     return background
 
-def show_img(all_imgs, label, bg, alpha):
+def show_img(mask_images, label, bg, alpha):
         
     idx = target_list_all.index(label)
 
-    mask_path = all_imgs[idx]["name"]
-    foreground = Image.open(mask_path).convert("RGBA")
-
+    foreground = mask_images[idx].convert("RGBA")
     background = bg.convert("RGBA")
 
     foreground.putalpha(int(255 * alpha))
@@ -209,15 +207,15 @@ with gr.Blocks(title=title) as app:
         dropdown = gr.Dropdown(choices=target_list_all, label="Select Label", value="All")
         slider = gr.Slider(minimum=0, maximum=1, value=0.4, label="Alpha Factor")
 
-    all_masks = gr.Gallery(visible=False) 
-    background = gr.Image(visible=False)
+    mask_state = gr.State() 
+    background_state = gr.State()
     
     gr.Button("1) Generate Masks").click(fn=inference, 
                                          inputs=[input_img],
-                                         outputs=[img, all_masks, background])
+                                         outputs=[img, mask_state, background_state])
 
     submit_transparent_img = gr.Button("2) Generate Transparent Mask (with Alpha Factor)")
-    submit_transparent_img.click(fn=show_img, inputs=[all_masks, dropdown, background, slider], outputs=[transparent_img])
+    submit_transparent_img.click(fn=show_img, inputs=[mask_state, dropdown, background_state, slider], outputs=[transparent_img])
    
      
 app.launch()