Update app_3.py

app_3.py

@@ -498,7 +498,7 @@ def run_rmbg(image):
     input_images = transform_image(image).unsqueeze(0).to("cuda")
     # Prediction
     with torch.no_grad():
-        preds = birefnet(input_images)[-1].sigmoid().cpu()
+        preds = rmbg(input_images)[-1].sigmoid().cpu()
     pred = preds[0].squeeze()
     pred_pil = transforms.ToPILImage()(pred)
     mask = pred_pil.resize(image_size)