Update app.py

app.py

@@ -32,7 +32,7 @@ def dilate_mask(mask, kernel_size=5, iterations=5):
     return Image.fromarray(mask)
 
 def remove_obj(image, seed):
-    alpha_channel = img["layers"][0][:, :, 3]
+    alpha_channel = image["layers"][0][:, :, 3]
     mask = np.where(alpha_channel == 0, 0, 255).astype(np.uint8)
     uploaded_mask = Image.fromarray(mask)
     background = Image.fromarray(img["background"])
@@ -93,4 +93,4 @@ with gr.Blocks(theme="gradio/monochrome") as demo:
 
 
 
-demo.launch()
+demo.launch(share=True)