I have imported the code from github and while running the code

!python /kaggle/working/Swin-TransUper/train.py

got error as

Traceback (most recent call last):
File "/kaggle/working/Swin-TransUper/train.py", line 10, in
from trainer import trainer_synapse,trainer_ISIC,trainer_busi,trainer_mix
File "/kaggle/working/Swin-TransUper/trainer.py", line 17, in
from utils import DiceLoss
File "/kaggle/working/Swin-TransUper/utils.py", line 3, in
from medpy import metric
ModuleNotFoundError: No module named 'medpy'

Where to set it within the code ?
assert inputs.size() == target.size(), 'predict {} & target {} shape do not match'.format(inputs.size(), target.size())
AssertionError: predict torch.Size([8, 1, 224, 224]) & target torch.Size([8, 2, 224, 224]) shape do not match