Update model.py

model.py

@@ -14,7 +14,7 @@ def create_vit_model(num_classes: int = 10,
   model = torchvision.models.vit_b_16(weights=weights)
 
   # 4. Freeze the base layers
-  for param in model.parameters()
+  for param in model.parameters():
     param.requires_grad = False
 
   # 5. Adjust the number of heads(output)