Update model.py

model.py

@@ -20,7 +20,7 @@ def create_mobilenet_model(num_classes:int=4,
     """
     # Create EffNetB2 pretrained weights, transforms and model
     
-    transforms = transforms.Compose([
+    transform = transforms.Compose([
         transforms.Resize((224, 224)), # 1. Reshape all images to 224x224 (though some models may require different sizes)
         transforms.ToTensor(), # 2. Turn image values to between 0 & 1
         transforms.Normalize(mean=[0.485, 0.456, 0.406], # 3. A mean of [0.485, 0.456, 0.406] (across each colour channel)
@@ -45,5 +45,5 @@ def create_mobilenet_model(num_classes:int=4,
                   bias=True)
     )
     
-    return model, transforms
+    return model, transform