Update model.py

model.py

@@ -17,7 +17,7 @@ def create_effnetb2_model(num_classes:int=3,  # default output class = 3
   # 5. Change classifier head with random seed for reproducability
   torch.manual_seed(seed)
   model.classifier = nn.Sequential(
-      nn.Dropout(p=0.3, inplace=True)
+      nn.Dropout(p=0.3, inplace=True),
       nn.Linear(in_features=1408, out_features=num_classes)
   )