Update model.py

model.py

@@ -4,7 +4,7 @@ import torch.nn as nn
 
 def create_effnetb2_model(num_classes:int=3, seed:int=3):
   weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
-  transforms = effnetb2_weights.transforms()
+  transforms = weights.transforms()
   model = torchvision.models.efficientnet_b2(weights=weights)
 
   # Freeze the base layers in the model (this will stop all layers from training)