Update encoder.py

encoder.py

@@ -7,7 +7,7 @@ class ImageEncoder(nn.Module):
     def __init__(self):
         super(ImageEncoder, self).__init__()
         # Load the Swin Transformer with features_only=True
-        self.swin = create_model("swin-tiny-patch4-window7-224 ", pretrained=True, features_only=True)
+        self.swin = create_model("swin-tiny-patch4-window7-224", pretrained=True, features_only=True)
         for param in self.swin.parameters():
             param.requires_grad = True