Update model.py

model.py

@@ -33,4 +33,4 @@ def create_effnetb2_model(num_classes:int=3,
         nn.Linear(in_features=1408, out_features=num_classes),
     )
     
-    return model, transforms
+    return model, transforms