Update model.py

model.py

@@ -1,7 +1,7 @@
 import torchvision
 from torch import nn
 
-def create_effnetb1(num_classes:int=5, seed:int=42):
+def create_effnet_b1(num_classes:int=5, seed:int=42):
     # 1. Get the base mdoel with pretrained weights and send to target device
     weights = torchvision.models.EfficientNet_B1_Weights.DEFAULT
     transforms = weights.transforms()