Update app.py

app.py

@@ -56,15 +56,15 @@ print("Model ready!")
 # Image preprocess
 ##################
 
-transform = transforms.Compose([
-    transforms.Resize((512, 512)),
-    transforms.ToTensor(),
-    transforms.ToPILImage(),
-    transforms.Normalize(
+to_tensor = transforms.ToTensor()
+to_array = transforms.ToPILImage()
+resize = transforms.Resize((512,512))
+resize_small = transforms.Resize((369,369))
+narmalize = transforms.Normalize 
+transform = transforms.Compose(
         mean=[0.485, 0.456, 0.406],
         std=[0.229, 0.224, 0.225]
     )
-])
 
 def process_pil(img):
     img = to_tensor(img)
@@ -110,7 +110,7 @@ def show_img(all_imgs, dropdown, bg, alpha_factor):
 ###########
 # Inference
 
-def inference(img, alpha_factor):
+def inference(img):
     background = resize_pil(img) 
 
     img = process_pil(img)
@@ -215,7 +215,7 @@ with gr.Blocks(title=title) as app:
     background = gr.Image(visible=False)
     
     gr.Button("1) Generate Masks").click(fn=inference, 
-                                         inputs=[input_img],
+                                         inputs=[input_img, slider],
                                          outputs=[img, all_masks, background])
 
     gr.Button("2) Generate Transparent Mask (with Alpha Factor)").click(fn=show_img,