fix

model/modelLoading.py

@@ -35,7 +35,7 @@ def loadBiSeNet(device: str = 'cpu') -> BiSeNet:
         model (BiSeNet): The loaded BiSeNet model.
     """
     model = BiSeNet(num_classes=19, context_path='resnet18').to(device)
-    model.load_state_dict(torch.load('./weights/BiSeNet/weightADV.pth', map_location=device))
+    model.load_state_dict(torch.load('./weights/BiSeNet/weightADV.pth', map_location=device)['model_state_dict'])
     model.eval()
     
     return model