Upload model.py

model.py

@@ -6,7 +6,7 @@ from torch import nn
 def create_effnetb2_model(num_classes:int=3,
                           seed:int=42):
     weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
-    transformers = weights.transformers()
+    transforms = weights.transforms()
     model = torchvision.models.efficientnet_b2(weights=weights)
 
     for param in model.parameters():
@@ -18,4 +18,4 @@ def create_effnetb2_model(num_classes:int=3,
         nn.Linear(in_features=1408, out_features=num_classes)
     )
 
-    return model, transformers
+    return model, transforms