fix typo

model.py

@@ -17,7 +17,7 @@ def create_effnetb2_model(num_classes: int = 3,
       A tuple (model, transforms) of the model and its image transforms.
     """
     weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
-    transforms = effnet_b2_weights.transforms()
+    transforms = weights.transforms()
     model = torchvision.models.efficientnet_b2(weights=weights)
     
     # Freeze parameters below the head