Update modnet_utils.py

modnet_utils.py

@@ -6,8 +6,8 @@ from model.modnet import MODNet
 
 def remove_background_modnet(image: Image.Image) -> Image.Image:
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-    modnet = MODNet(backbone_pretrained=False)
-    modnet.load_state_dict(torch.load('pretrained/modnet_photographic_portrait_matting.ckpt', map_location=device))
+    modnet = MODNet(backbone='mobilenetv2', backbone_pretrained=False)
+    modnet.load_state_dict(torch.load('pretrained/modnet_webcam_portrait_matting.ckpt', map_location=device))
     modnet = modnet.to(device).eval()
 
     img = np.array(image.convert("RGB"))