Update model.py

model.py

@@ -7,12 +7,10 @@ from torch import nn
 def create_effnetb2_model(num_classes:int=3, 
                           seed:int=42):
     """Creates an EfficientNetB2 feature extractor model and transforms.
-
     Args:
         num_classes (int, optional): number of classes in the classifier head. 
             Defaults to 3.
         seed (int, optional): random seed value. Defaults to 42.
-
     Returns:
         model (torch.nn.Module): EffNetB2 feature extractor model. 
         transforms (torchvision.transforms): EffNetB2 image transforms.