Update modnet_utils.py

modnet_utils.py

@@ -6,7 +6,12 @@ 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='mobilenetv2')
+    modnet = MODNet()
+    from model.backbones import SUPPORTED_BACKBONES  # если ещё не импортировано
+    backbone = SUPPORTED_BACKBONES 
+    modnet.backbone = backbone
+    modnet.backbone_arch = 'mobilenetv2'
+
     modnet.load_state_dict(torch.load('pretrained/modnet_webcam_portrait_matting.ckpt', map_location=device))
     modnet.to(device)
     modnet.eval()