change the model classifiler to suit b6

model.py

@@ -29,8 +29,8 @@ def create_effnetb2_model(num_classes:int=3,
     # Change classifier head with random seed for reproducibility
     torch.manual_seed(seed)
     model.classifier = nn.Sequential(
-        nn.Dropout(p=0.3, inplace=True),
-        nn.Linear(in_features=1408, out_features=num_classes),
+        nn.Dropout(p=0.5, inplace=True),
+        nn.Linear(in_features=2304, out_features=num_classes),
     )
     
     return model, transforms