diff --git a/inceptionnet_finetune.py b/inceptionnet_finetune.py
index 25c9d2b..4f5065e 100644
--- a/inceptionnet_finetune.py
+++ b/inceptionnet_finetune.py
@@ -1,5 +1,6 @@
 from os import makedirs
 from torchvision.models import Inception_V3_Weights, inception_v3
+from torchvision.models import swin_v2_b, Swin_V2_B_Weights
 from torch.nn import Linear
 from torchvision.transforms import Compose, RandomResizedCrop, RandomRotation, ToTensor, \
     RandomHorizontalFlip, \